commit | f6c457cdc27f232d7cf7f4f9b801aa8c3f9f18c6 | [log] [tgz] |
---|---|---|
author | Googler <noreply@google.com> | Tue Mar 26 11:14:22 2024 -0700 |
committer | Copybara-Service <copybara-worker@google.com> | Tue Mar 26 11:15:05 2024 -0700 |
tree | 5226b6207d50c4abdd9b143d8187be75991adfb0 | |
parent | 24b67b333c27d568c44af32af4f8d5a99e032dfe [diff] |
No public description PiperOrigin-RevId: 619251887 Change-Id: Ia819717250bc88294af24451d3340822e72c429f
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h index ea34035..ef553e0 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionGpu.h
@@ -1057,6 +1057,8 @@ __syncthreads(); } // end loop over k + #undef add_vals + __syncthreads(); Index horiz_base = (threadIdx.y / 4) * 8 + base_n; if (!CHECK_LHS_BOUNDARY && !CHECK_RHS_BOUNDARY) {