[tvm] branch main updated: fixed cutlass byoc build break (#11686)

[masahi]

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

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


The following commit(s) were added to refs/heads/main by this push:
     new 005f05e26f fixed cutlass byoc build break (#11686)
005f05e26f is described below

commit 005f05e26f895c12b522e3e770884cc98d7e54b9
Author: Masahiro Masuda <masahi...@gmail.com>
AuthorDate: Mon Jun 13 19:02:27 2022 +0900

    fixed cutlass byoc build break (#11686)
---
 src/relay/backend/contrib/cutlass/codegen.cc | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/src/relay/backend/contrib/cutlass/codegen.cc 
b/src/relay/backend/contrib/cutlass/codegen.cc
index 97c039ee29..772007792a 100644
--- a/src/relay/backend/contrib/cutlass/codegen.cc
+++ b/src/relay/backend/contrib/cutlass/codegen.cc
@@ -602,7 +602,8 @@ class CodegenCutlass : public 
backend::MemoizedExprTranslator<std::vector<Output
     ICHECK(pattern_name.defined()) << "Only functions with composite attribute 
are supported.";
 
     if (pattern_name == "cutlass.dense") {
-      const auto* dense_call = GetRootCall(callee->body.as<CallNode>(), 0, 
"nn.dense");
+      const auto* dense_call =
+          GetRootCall(callee->body.as<CallNode>(), 0, 
std::vector<std::string>{"nn.dense"});
       return GenerateBody(dense_call, "cutlass_dense", 
GetArgumentNames(caller),
                           DenseArgs(std::ref(attrs_)));
     } else if (pattern_name == "cutlass.dense_bias") {
@@ -637,11 +638,12 @@ class CodegenCutlass : public 
backend::MemoizedExprTranslator<std::vector<Output
                           DenseArgs(std::ref(attrs_)));
     } else if (pattern_name == "cutlass.batch_matmul") {
       const auto* batch_matmul_call =
-          GetRootCall(callee->body.as<CallNode>(), 0, "nn.batch_matmul");
+          GetRootCall(callee->body.as<CallNode>(), 0, 
std::vector<std::string>{"nn.batch_matmul"});
       return GenerateBody(batch_matmul_call, "cutlass_batch_matmul", 
GetArgumentNames(caller),
                           BatchMatmulArgs(std::ref(attrs_)));
     } else if (pattern_name == "cutlass.conv2d") {
-      const auto* conv2d_call = GetRootCall(callee->body.as<CallNode>(), 0, 
"nn.conv2d");
+      const auto* conv2d_call =
+          GetRootCall(callee->body.as<CallNode>(), 0, 
std::vector<std::string>{"nn.conv2d"});
       return GenerateBody(conv2d_call, "cutlass_conv2d", 
GetArgumentNames(caller),
                           Conv2dArgs(std::ref(attrs_)));
     } else if (pattern_name == "cutlass.conv2d_bias") {
@@ -704,12 +706,13 @@ class CodegenCutlass : public 
backend::MemoizedExprTranslator<std::vector<Output
       return GenerateBody(conv2d_call, pattern_name.value(), 
GetArgumentNames(caller),
                           Conv2dArgs(std::ref(attrs_)));
     } else if (pattern_name == "cutlass.conv2d_transpose") {
-      const auto* conv2d_call = GetRootCall(callee->body.as<CallNode>(), 0, 
"nn.conv2d_transpose");
+      const auto* conv2d_call = GetRootCall(callee->body.as<CallNode>(), 0,
+                                            
std::vector<std::string>{"nn.conv2d_transpose"});
       return GenerateBody(conv2d_call, "cutlass_conv2d_transpose", 
GetArgumentNames(caller),
                           Conv2dArgs(std::ref(attrs_), true, false));
     } else if (pattern_name == "cutlass.conv2d_backward_weight") {
-      const auto* conv2d_call =
-          GetRootCall(callee->body.as<CallNode>(), 0, 
"nn.conv2d_backward_weight");
+      const auto* conv2d_call = GetRootCall(callee->body.as<CallNode>(), 0,
+                                            
std::vector<std::string>{"nn.conv2d_backward_weight"});
       return GenerateBody(conv2d_call, "cutlass_conv2d_backward_weight", 
GetArgumentNames(caller),
                           Conv2dArgs(std::ref(attrs_), false, true));
     }