diff --git a/python/tvm/contrib/cutlass/gen_tensor_op.py b/python/tvm/contrib/cutlass/gen_tensor_op.py index 177304e10229..1de193858036 100644 --- a/python/tvm/contrib/cutlass/gen_tensor_op.py +++ b/python/tvm/contrib/cutlass/gen_tensor_op.py @@ -94,6 +94,7 @@ def generate_sm50_simt( DataType.f32, DataType.f32, DataType.f32, + DataType.f32, OpcodeClass.Simt, MathOperation.multiply_add, ),