Set ASEM to DepthU for PackSummationDims=1
@@ -1696,6 +1696,12 @@ class Solution:
state["VectorWidth"]))
return
+ if state["PackSummationDims"] == 1:
+ if state["DepthU"] % state["AssertSummationElementMultiple"] != 0:
+ reject(state, "PackSummationDims=1 requires DepthU is integer multiple of ASEM")
+ else:
+ state["AssertSummationElementMultiple"] = state["DepthU"]
+
# Some restrictions for half:
if state["KernelLanguage"] == "Assembly" \
and state["ProblemType"]["DataType"].isHalf():
