OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 12:20:11 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #48283 from reedwm/cherrypick-sparse-to-dense

Browse Source 
2.5-rc1 cherry-pick request: Disable GPU SparseToDense
This commit is contained in:
Mihai Maruseac 2021-04-09 13:09:20 -07:00 committed by GitHub
commit fadc35d749
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/kernels/sparse_to_dense_op.cc
View File

@ -273,6 +273,9 @@ class SparseToDenseGPU : public AsyncOpKernel {
bool validate_indices_; bool validate_indices_;
}; };
// TODO(b/184077412): SparseToDense causes an illegal access error.
#if 0
#define REGISTER_GPU_KERNELS(type, index_type) \ #define REGISTER_GPU_KERNELS(type, index_type) \
REGISTER_KERNEL_BUILDER(Name("SparseToDense") \ REGISTER_KERNEL_BUILDER(Name("SparseToDense") \
.Device(DEVICE_GPU) \ .Device(DEVICE_GPU) \
@ -292,6 +295,7 @@ REGISTER_GPU_KERNELS_ALL(bool)
#undef REGISTER_GPU_KERNELS_ALL #undef REGISTER_GPU_KERNELS_ALL
#undef REGISTER_GPU_KERNELS #undef REGISTER_GPU_KERNELS
#endif
#endif // GOOGLE_CUDA #endif // GOOGLE_CUDA

2
tensorflow/python/kernel_tests/BUILD
View File

@ -1121,7 +1121,7 @@ tf_py_test(
], ],
) )
cuda_py_test( tf_py_test(
name = "sparse_to_dense_op_py_test", name = "sparse_to_dense_op_py_test",
size = "small", size = "small",
srcs = ["sparse_to_dense_op_py_test.py"], srcs = ["sparse_to_dense_op_py_test.py"],