fp16/bf16 solutions utilizing matrix instructions should have HPA properly enabled
@@ -1267,6 +1267,11 @@ class Solution:
state["MatrixInstN"] = state["MatrixInstruction"][1]
state["MatrixInstK"] = state["MatrixInstruction"][2]
state["MatrixInstB"] = state["MatrixInstruction"][3]
+ if not state["ProblemType"]["HighPrecisionAccumulate"] and \
+ not state["ProblemType"]["DataType"].isSingle() :
+ reject(state, "Matrix instructions for half types are natively accumulated" + \
+ " in fp32 precision. Please add the following config:" + \
+ "\n - HighPrecisionAccumulate: True")
else:
if state["ThreadTile"][0] > 16 or state["ThreadTile"][1] > 16:
reject(state, "Invalid value for ThreadTile")
