Commit dfbe82b0 by KeDengMS Committed by Tianqi Chen

Fixes for tensorize in Windows build to expose TensorIntrin::make and search clang.exe (#1896)

parent 10279098
...@@ -74,13 +74,13 @@ class TensorIntrinNode : public Node { ...@@ -74,13 +74,13 @@ class TensorIntrinNode : public Node {
v->Visit("reduce_update", &reduce_update); v->Visit("reduce_update", &reduce_update);
} }
static TensorIntrin make(std::string name, TVM_DLL static TensorIntrin make(std::string name,
Operation op, Operation op,
Array<Tensor> inputs, Array<Tensor> inputs,
Array<Buffer> buffers, Array<Buffer> buffers,
Stmt body, Stmt body,
Stmt reduce_init, Stmt reduce_init,
Stmt reduce_update); Stmt reduce_update);
static constexpr const char* _type_key = "TensorIntrin"; static constexpr const char* _type_key = "TensorIntrin";
TVM_DECLARE_NODE_TYPE_INFO(TensorIntrinNode, Node); TVM_DECLARE_NODE_TYPE_INFO(TensorIntrinNode, Node);
...@@ -90,7 +90,6 @@ inline const TensorIntrinNode* TensorIntrin::operator->() const { ...@@ -90,7 +90,6 @@ inline const TensorIntrinNode* TensorIntrin::operator->() const {
return static_cast<const TensorIntrinNode*>(node_.get()); return static_cast<const TensorIntrinNode*>(node_.get());
} }
// Internal node container of tensor intrinsic calling. // Internal node container of tensor intrinsic calling.
class TensorIntrinCallNode; class TensorIntrinCallNode;
......
...@@ -31,6 +31,7 @@ def find_clang(required=True): ...@@ -31,6 +31,7 @@ def find_clang(required=True):
if hasattr(codegen, "llvm_version_major"): if hasattr(codegen, "llvm_version_major"):
cc_list += ["clang-%d.0" % codegen.llvm_version_major()] cc_list += ["clang-%d.0" % codegen.llvm_version_major()]
cc_list += ["clang"] cc_list += ["clang"]
cc_list += ["clang.exe"]
valid_list = [util.which(x) for x in cc_list] valid_list = [util.which(x) for x in cc_list]
valid_list = [x for x in valid_list if x] valid_list = [x for x in valid_list if x]
if not valid_list and required: if not valid_list and required:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment