You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by tq...@apache.org on 2023/03/22 22:53:07 UTC

[tvm] branch unity updated: [Unity] Also include output dtype in simt MathInstruction (#14372)

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/unity by this push:
     new 4e46ad4e9c [Unity] Also include output dtype in simt MathInstruction (#14372)
4e46ad4e9c is described below

commit 4e46ad4e9c1662a346fdf69b6cfdac1e7687e8c0
Author: Andrew Reusch <ar...@gmail.com>
AuthorDate: Wed Mar 22 15:53:00 2023 -0700

    [Unity] Also include output dtype in simt MathInstruction (#14372)
    
    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 177304e102..1de1938580 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,
             ),