From dfbe82b045072356fa78a3c6b0a87b83543c661c Mon Sep 17 00:00:00 2001
From: KeDengMS <kedeng@microsoft.com>
Date: Fri, 12 Oct 2018 14:20:37 -0700
Subject: [PATCH] Fixes for tensorize in Windows build to expose
TensorIntrin::make and search clang.exe (#1896)
---
include/tvm/tensor_intrin.h | 15 +++++++--------
python/tvm/contrib/clang.py | 1 +
2 files changed, 8 insertions(+), 8 deletions(-)
diff --git a/include/tvm/tensor_intrin.h b/include/tvm/tensor_intrin.h
index fbee4bccc..6cffc931d 100644
--- a/include/tvm/tensor_intrin.h
+++ b/include/tvm/tensor_intrin.h
@@ -74,13 +74,13 @@ class TensorIntrinNode : public Node {
v->Visit("reduce_update", &reduce_update);
}
- static TensorIntrin make(std::string name,
- Operation op,
- Array<Tensor> inputs,
- Array<Buffer> buffers,
- Stmt body,
- Stmt reduce_init,
- Stmt reduce_update);
+ TVM_DLL static TensorIntrin make(std::string name,
+ Operation op,
+ Array<Tensor> inputs,
+ Array<Buffer> buffers,
+ Stmt body,
+ Stmt reduce_init,
+ Stmt reduce_update);
static constexpr const char* _type_key = "TensorIntrin";
TVM_DECLARE_NODE_TYPE_INFO(TensorIntrinNode, Node);
@@ -90,7 +90,6 @@ inline const TensorIntrinNode* TensorIntrin::operator->() const {
return static_cast<const TensorIntrinNode*>(node_.get());
}
-
// Internal node container of tensor intrinsic calling.
class TensorIntrinCallNode;
diff --git a/python/tvm/contrib/clang.py b/python/tvm/contrib/clang.py
index 19508160d..3e8ad663c 100644
--- a/python/tvm/contrib/clang.py
+++ b/python/tvm/contrib/clang.py
@@ -31,6 +31,7 @@ def find_clang(required=True):
if hasattr(codegen, "llvm_version_major"):
cc_list += ["clang-%d.0" % codegen.llvm_version_major()]
cc_list += ["clang"]
+ cc_list += ["clang.exe"]
valid_list = [util.which(x) for x in cc_list]
valid_list = [x for x in valid_list if x]
if not valid_list and required:
--
GitLab