gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[python] fixed warnings for pybind11 and pytorch

Browse Source
This commit is contained in:
Philippe Tillet
2019-09-05 20:28:00 -04:00
parent 3fd61c1a02
commit 1f8fd525b5
3 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/runtime/function.cc
View File

@@ -220,8 +220,6 @@ std::unique_ptr<driver::module> function::make_bin(ir::module &module, driver::c
dce.run(module); dce.run(module);
vectorize.run(module); vectorize.run(module);
dce.run(module); dce.run(module);
// generate llvm code // generate llvm code
llvm::LLVMContext ctx; llvm::LLVMContext ctx;
std::unique_ptr<llvm::Module> llvm(new llvm::Module(module.get_name(), ctx)); std::unique_ptr<llvm::Module> llvm(new llvm::Module(module.get_name(), ctx));

2
python/src/bindings.cc
View File

@@ -493,7 +493,7 @@ extern std::map<size_t, std::shared_ptr<rt::function>> id_fn_map;
oss << std::endl; oss << std::endl;
oss << std::endl; oss << std::endl;
oss << "static auto registry = torch::jit::RegisterOperators(\"triton::" << name << "\", &" << name << ");" << std::endl; oss << "static auto registry = torch::RegisterOperators(\"triton::" << name << "\", &" << name << ");" << std::endl;
return {oss.str(), name}; return {oss.str(), name};
} }

24
python/src/pybind11/functional.h
View File

@@ -65,12 +65,26 @@ public:
} }
}; };
value = [hfunc = func_handle(std::move(func))](Args... args) -> Return { // value = [hfunc = func_handle(std::move(func))](Args... args) -> Return {
gil_scoped_acquire acq; // gil_scoped_acquire acq;
object retval(hfunc.f(std::forward<Args>(args)...)); // object retval(hfunc.f(std::forward<Args>(args)...));
/* Visual studio 2015 parser issue: need parentheses around this expression */ // /* Visual studio 2015 parser issue: need parentheses around this expression */
return (retval.template cast<Return>()); // return (retval.template cast<Return>());
// };
struct func_wrapper {
func_handle hfunc;
func_wrapper(func_handle&& hf): hfunc(std::move(hf)) {}
Return operator()(Args... args) const {
gil_scoped_acquire acq;
object retval(hfunc.f(std::forward<Args>(args)...));
/* Visual studio 2015 parser issue: need parentheses around this expression */
return (retval.template cast<Return>());
}
}; };
value = func_wrapper(func_handle(std::move(func)));
return true; return true;
} }