experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Correct #if

Browse Source
This commit is contained in:
Eugene Kuznetsov 2020-03-11 14:33:23 -07:00
parent 6aa431f8c9
commit 8f85a119c2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/grappler/utils_test.cc
View File

@ -423,7 +423,7 @@ TEST(IsKernelRegisteredForNode, All) {
v.set_type(DataType::DT_FLOAT);
(*node.mutable_attr())["T"] = v;
TF_EXPECT_OK(IsKernelRegisteredForNode(node));
#ifdef GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
node.set_device("/gpu:0");
TF_EXPECT_OK(IsKernelRegisteredForNode(node));
#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM