You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by ma...@apache.org on 2022/06/13 10:02:36 UTC
[tvm] branch main updated: fixed cutlass byoc build break (#11686)
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 <ma...@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));
}