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

Fix for build without ROCm or CUDA

Browse Source
This commit is contained in:
Eugene Kuznetsov 2020-01-17 03:18:04 -08:00
parent 3122804c85
commit b84908da77
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/stateful_random_ops_cpu_gpu.h
View File

@ -82,10 +82,6 @@ struct RngSkip_Philox;
using CPUDevice = Eigen::ThreadPoolDevice;
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
using GPUDevice = Eigen::GpuDevice;
struct UpdateVariableAndFill_Philox_Arg {
int64 output_size;
int64 alg_tag_skip;
@ -93,6 +89,10 @@ struct UpdateVariableAndFill_Philox_Arg {
Tensor* state_tensor;
};
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
using GPUDevice = Eigen::GpuDevice;
// Declares the partially GPU-specialized functor structs.
// must be kept at <=6 arguments because of a gcc/clang ABI incompatibility bug
template <typename Distribution>