experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Step two of renaming some files from cuda_* to gpu_*: Change #include directives. See PR #24293.

Browse Source 
PiperOrigin-RevId: 243143199
This commit is contained in:
A. Unique TensorFlower 2019-04-11 14:31:27 -07:00 committed by TensorFlower Gardener
parent 4accf25bb1
commit ea3cb637ac
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/examples/adding_an_op/cuda_op_kernel.cu.cc
View File

@ -16,7 +16,7 @@ limitations under the License.
#if GOOGLE_CUDA
#define EIGEN_USE_GPU
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/util/cuda_launch_config.h"
#include "tensorflow/core/util/gpu_launch_config.h"
__global__ void AddOneKernel(const int* in, const int N, int* out) {
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < N;

2
tensorflow/tools/ci_build/builds/user_ops/cuda_op_kernel.cu.cc
View File

@ -16,7 +16,7 @@ limitations under the License.
#if GOOGLE_CUDA
#define EIGEN_USE_GPU
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/util/cuda_launch_config.h"
#include "tensorflow/core/util/gpu_launch_config.h"
__global__ void AddOneKernel(const int* in, const int N, int* out) {
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < N;