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

[dnn] better specification of recompilation key

Browse Source
This commit is contained in:
Philippe Tillet
2019-08-02 17:42:48 -07:00
parent 3b92ddf7e6
commit d9945692a9
31 changed files with 418 additions and 428 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/python/pytorch/shift.cpp
View File

@@ -49,9 +49,9 @@ torch::Tensor shift_common(
std::string dtype;
at::ScalarType type = torcha.scalar_type();
switch(type){
case at::ScalarType::Double: dtype = "fp64"; break;
case at::ScalarType::Float: dtype = "fp32"; break;
case at::ScalarType::Half: dtype = "fp16"; break;
case at::ScalarType::Double: dtype = "double"; break;
case at::ScalarType::Float: dtype = "float"; break;
case at::ScalarType::Half: dtype = "half"; break;
default: AT_ERROR("unknown data-type for shift-conv");
}
// Get configuration