lukeiwanski / tensorflow

OpenCL support for TensorFlow via SYCL
Apache License 2.0
65 stars 14 forks source link

[OpenCL] Fixes SYCL registration of ScatterNd op #120

Closed jwlawson closed 7 years ago

jwlawson commented 7 years ago

The changes to the ScatterNd op in 63ac86a placed the SYCl kernel registration inside GOOGLE_CUDA guards, which means that they weren't registered except when compiled with both CUDA and SYCL support. Here I ensure that the SYCL registrations are outside of the GOOGLE_CUDA scope.