Skip to content

Commit cbcdc6e

Browse files
Luke Iwanskibenoitsteiner
authored andcommitted
Sycl improvements (#44)
- Eigen version bump - Extends Cast and Cwise ops benchmark to cover Sycl device - Extends device_lib_test.py to cover Sycl device - Registers int32, string and ResourceHandler to run on host for Enter and RefEnter Sycl Ops - Enables RecudeMax op for Sycl since Eigen implementation is ready - Registers Less op for Sycl device
1 parent 7ce6097 commit cbcdc6e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/core/kernels/reduction_ops_max.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ REGISTER_KERNEL_BUILDER(
6666
.TypeConstraint<int32>("Tidx") \
6767
.HostMemory("reduction_indices"), \
6868
ReductionOp<SYCLDevice, type, Eigen::internal::MaxReducer<type>>);
69-
// REGISTER_SYCL_KERNELS(float);
69+
REGISTER_SYCL_KERNELS(float);
7070
#undef REGISTER_SYCL_KERNELS
7171

7272
REGISTER_KERNEL_BUILDER(

0 commit comments

Comments
 (0)