Remove TT shape restrictions. Testing shows they are not needed.
@@ -1406,9 +1406,8 @@ class Solution:
# For now, LocalDotLayout > 1 only works if the thread tile is a square and VectorWidth is equal to the
# thread tile size
ldl = state["LocalDotLayout"]
- if ldl > 1 and \
- (state["ThreadTile0"] != state["VectorWidth"] or state["ThreadTile1"] != state["VectorWidth"] or state["AssertSummationElementMultiple"] % ldl != 0):
- reject(state, "LocalDotLayout > 1 only supports square thread tiles and VectorWidth equal to ThreadTile0/1 size and ASEM a multiple of LDL")
+ if ldl > 1 and (state["AssertSummationElementMultiple"] % ldl != 0):
+ reject(state, "LocalDotLayout > 1 only supports ASEM a multiple of LDL")
return
if 0:
