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 {
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;
......
......@@ -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:
......
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