diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-05 11:57:53 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-05 12:00:50 -0800 |
| commit | 21e831dc4a3fe67331dc186d2493678f24838250 (patch) | |
| tree | beb80405062cb1804401944bac6930b182108c91 /tensorflow/core/kernels/svd_op_gpu.cu.cc | |
| parent | 33e3da538aef02b4f1293969d9601cc98f3ad30c (diff) | |
Automated g4 rollback of changelist 177799252
PiperOrigin-RevId: 177989542
Diffstat (limited to 'tensorflow/core/kernels/svd_op_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/svd_op_gpu.cu.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/svd_op_gpu.cu.cc b/tensorflow/core/kernels/svd_op_gpu.cu.cc index 8c3a58b108..dedc2da60b 100644 --- a/tensorflow/core/kernels/svd_op_gpu.cu.cc +++ b/tensorflow/core/kernels/svd_op_gpu.cu.cc @@ -63,8 +63,8 @@ __global__ void ComputeValueOfVKernel(Cuda2DLaunchConfig config, int64 m, int64 ldu, const Scalar* M, const Scalar* U, const Scalar* S, Scalar* V) { - CUDA_AXIS_KERNEL_LOOP(batch, config.virtual_thread_count.x, X) { - CUDA_AXIS_KERNEL_LOOP(i, config.virtual_thread_count.y, Y) { + CUDA_AXIS_KERNEL_LOOP(batch, config.virtual_thread_count, x) { + CUDA_AXIS_KERNEL_LOOP(i, config.virtual_thread_count, y) { Scalar v = M[i + m * batch] * U[ldu * (i + m * batch)] * S[batch]; CudaAtomicAdd(V + batch, v); } |