Skip to content

Commit b00431d

Browse files
Luke Iwanskibenoitsteiner
authored andcommitted
[OpenCL] Fixes an issue caused by switch to aligned allocator for sycl buffer (#53)
1 parent aa2d4f4 commit b00431d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/core/kernels/random_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -695,7 +695,7 @@ void FillPhiloxRandom<SYCLDevice, Distribution>::operator()(
695695
const size_t group_size = device.maxSyclThreadsPerBlock();
696696
const size_t group_count = (size + group_size - 1) / group_size;
697697

698-
sycl::buffer<uint8_t, 1> buffer = device.get_sycl_buffer(data);
698+
auto buffer = device.get_sycl_buffer(data);
699699

700700
device.sycl_queue().submit([&](sycl::handler& cgh) {
701701
auto access = buffer.template get_access<sycl::access::mode::write>(cgh);

0 commit comments

Comments
 (0)