From ead762f31c0bde4145ee39457783472d6458352d Mon Sep 17 00:00:00 2001 From: daineAMD Date: Wed, 4 Dec 2024 11:09:36 -0700 Subject: [PATCH] Add reject states for failing streamk params. --- tensilelite/Tensile/SolutionStructs.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tensilelite/Tensile/SolutionStructs.py b/tensilelite/Tensile/SolutionStructs.py index db1809b8f7..57518a1f8a 100644 --- 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")