experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/core/kernels/sparse
History
Deven Desai 5bf0bab331 [ROCm] Fix for ROCm CSB breakage on 200507 
The following PR/commit introduces a build error on the ROCm platform

https://github.com/tensorflow/tensorflow/pull/38802

The error is caused by a call to the `CsrgemmBufferSize` routine which only exists on the CUDA side. The call to it was not guarded by the same #if block that guards the function declaration + definition. Adding the missing #if block fixes the issue.

This PR also adds some explicit `GOOGLE_CUDA &&` and `|| TENSORFLOW_USE_ROCM` conditions to some `#if` to make things clear.
2020-05-07 20:55:49 +00:00
..
add_op.cc
Merge pull request #38802 from nluehr:cusparse_remove_deprecated
2020-05-06 11:05:02 -07:00
BUILD
Merge branch 'master' into google-upstream-gpuprim
2020-03-16 12:50:58 -07:00
conj_op.cc
…
csr_sparse_matrix_to_dense_op.cc
Complex sparse ops
2020-01-16 19:35:32 -08:00
csr_sparse_matrix_to_sparse_tensor_op.cc
…
dense_to_csr_sparse_matrix_op.cc
Merge pull request #35966 from ROCmSoftwarePlatform:google-upstream-sparse-complex
2020-03-27 11:24:26 -07:00
kernels_gpu.cu.cc
Merge pull request #36640 from ROCmSoftwarePlatform:google-upstream-gpuprim
2020-04-15 09:18:52 -07:00
kernels_test.cc
…
kernels.cc
…
kernels.h
Cleanup formatting with clang-format
2020-05-04 17:40:14 -05:00
mat_mul_op.cc
[ROCm] Fix for ROCm CSB breakage on 200507
2020-05-07 20:55:49 +00:00
mul_op.cc
…
nnz_op.cc
…
softmax_op.cc
Adding ROCm support for the CSR Sparse Matrix Ops
2019-12-13 15:24:42 +00:00
sparse_cholesky_op.cc
…
sparse_mat_mul_op.cc
[ROCm] Fix for ROCm CSB breakage on 200507
2020-05-07 20:55:49 +00:00
sparse_matrix_components_op.cc
…
sparse_matrix.cc
…
sparse_matrix.h
…
sparse_ordering_amd_op.cc
…
sparse_tensor_to_csr_sparse_matrix_op.cc
Merge pull request #35966 from ROCmSoftwarePlatform:google-upstream-sparse-complex
2020-03-27 11:24:26 -07:00
transpose_op.cc
Add macros to avoid frequent ROCm ifdefs. Clean up the ifdefs in core/kernels/sparse.
2020-05-05 13:02:34 -07:00
transpose_op.h
Add CSRSparseMatrix ops.
2019-09-03 16:49:47 -07:00
zeros_op.cc
…
zeros_op.h
…