From 100ecb85f59ba6b314d770926b0bbdd1afa91dcc Mon Sep 17 00:00:00 2001
From: Andrew Reusch <areusch@gmail.com>
Date: Wed, 22 Mar 2023 13:54:25 -0700
Subject: [PATCH] Also include output dtype in simt MathInstruction.

---
 python/tvm/contrib/cutlass/gen_tensor_op.py | 1 +
 1 file changed, 1 insertion(+)

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,
             ),