Add scheuldeIteralg == 1 as valid for streamk.

ROCm · Dec 20, 2024 · 2229174 · 2229174
1 parent 318753a
commit 2229174
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensilelite/Tensile/SolutionStructs.py b/tensilelite/Tensile/SolutionStructs.py
@@ -2185,7 +2185,7 @@ def assignDerivedParameters(state):
         reject(state, "ScheduleGlobalRead not supported with Stream-K")
       if state["ScheduleLocalWrite"] != 1:
         reject(statue, "ScheduleLocalWrite not supported with Stream-K")
-      if state["ScheduleIterAlg"] != 3:
+      if state["ScheduleIterAlg"] != 1 and state["ScheduleIterAlg"] != 3:
         reject(state, "ScheduleIterAlg not supported with Stream-K")
       if state["StreamKAtomic"] == 1:
         if not state["ProblemType"]["DataType"].isSingle():