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

[codegen] fixed issue in double buffering pointer update

Browse Source
This commit is contained in:
Philippe Tillet
2019-08-28 17:50:45 -07:00
parent 59281f5794
commit d457482539
9 changed files with 135 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/src/tensorflow.cc
View File

@@ -160,7 +160,8 @@ void gen_register_op(std::ostream &os, const std::string &name,
std::string name = arg->get_name();
auto tolower = [](char c) { return std::tolower(c);};
std::transform(name.begin(), name.end(), name.begin(), tolower);
os << " .Input(\"" << name << ": " << to_tf_scalar_ty(arg->get_type()) << "\")\n";
os << " .Attr(\"T" << i << " : {bool, int8, int16, int32, int64, float16, float32, float64}\")" << std::endl;
os << " .Input(\"" << name << ": T" << i << "\")\n";
}
for(size_t i = 0; i < outputs.size(); i++){
std::string name = outputs[i];