Add reject states for failing streamk params.

ROCm · Dec 4, 2024 · ead762f · ead762f
1 parent 8ddfcab
commit ead762f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tensilelite/Tensile/SolutionStructs.py b/tensilelite/Tensile/SolutionStructs.py
@@ -2169,6 +2169,12 @@ def assignDerivedParameters(state):
         reject(state, "General batch not supported with Stream-K")
       if state["ProblemType"]["GroupedGemm"]:
         reject(state, "Grouped gemm not yet supported with Stream-K")
+      if state["ScheduleGlobalRead"] != 1:
+        reject(state, "ScheduleGlobalRead not supported with Stream-K")
+      if state["ScheduleLocalWrite"] != 1:
+        reject(statue, "ScheduleLocalWrite not supported with Stream-K")
+      if state["ScheduleIterAlg"] != 3:
+        reject(state, "ScheduleIterAlg not supported with Stream-K")
       if state["StreamKAtomic"] == 1:
         if not state["ProblemType"]["DataType"].isSingle():
           reject(state, "Atomic Stream-K currently only tested for SGEMM")