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
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 << "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};
}