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

[PYTHON][SRC][BINDING] Improved code portability across compilers

Browse Source
This commit is contained in:
Philippe Tillet
2020-04-17 13:51:47 -04:00
committed by Philippe Tillet
parent c36ad6bf8a
commit f805ff278a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/src/bindings.cc
View File

@@ -408,7 +408,7 @@ oss << R"(
)"; )";
gen_tf_register_op(oss, cc_name, args, allocs); gen_tf_register_op(oss, cc_name, args, allocs);
return {oss.str(), name}; return std::tuple<std::string, std::string>{oss.str(), name};
} }
@@ -598,7 +598,7 @@ extern std::map<size_t, int64_t> i64scalar_map;
oss << std::endl; oss << std::endl;
oss << "static auto registry = torch::RegisterOperators(\"triton::" << name << "\", &" << name << ");" << std::endl; oss << "static auto registry = torch::RegisterOperators(\"triton::" << name << "\", &" << name << ");" << std::endl;
return {oss.str(), name}; return std::tuple<std::string, std::string>{oss.str(), name};
} }
/* Function signature */ /* Function signature */