OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

DO NOT MERGE: debugging ROCm failure

Browse Source 
ghstack-source-id: 9391899c34
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34091
This commit is contained in:
Xiang Gao 2020-03-02 17:57:11 -08:00
parent b6b2ef20b3
commit 6f6ee5cdb1
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/native/cuda/DistributionBernoulli.cu
View File

@ -41,7 +41,7 @@ void bernoulli_tensor_cuda_kernel(
at::native::gpu_kernel(iter,
[seeds] GPU_LAMBDA (prob_t p) -> scalar_t {
#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__)
#if defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_HCC__)
curandStatePhilox4_32_10_t state;
curand_init(
seeds.first,

2
aten/src/ATen/native/cuda/Distributions.cu
View File

@ -84,7 +84,7 @@ void gamma_cuda_kernel(
at::native::gpu_kernel(iter,
[seeds] GPU_LAMBDA (scalar_t alpha) {
#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__)
#if defined(__CUDA_ARCH__) || defined(__HIP_PLATFORM_HCC__)
curandStatePhilox4_32_10_t state;
curand_init(
seeds.first,