From 19e343b18403cebc99b7f524f002dddf913a68e3 Mon Sep 17 00:00:00 2001 From: Johannes M Dieterich Date: Tue, 11 Sep 2018 16:00:00 -0500 Subject: [PATCH 1/2] Correct the warp size for current AMD GPUs. --- aten/src/THCUNN/LookupTableBag.cu | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/aten/src/THCUNN/LookupTableBag.cu b/aten/src/THCUNN/LookupTableBag.cu index c2ba9f52081972..83a60efbb32207 100644 --- a/aten/src/THCUNN/LookupTableBag.cu +++ b/aten/src/THCUNN/LookupTableBag.cu @@ -15,7 +15,11 @@ #include "THCHalfAutoNumerics.cuh" #include "THCTensorSort.cuh" +#if defined(__HIP_PLATFORM_HCC__) +const int WARP_SIZE = 64; +#else const int WARP_SIZE = 32; +#endif const int MODE_SUM = 0; const int MODE_MEAN = 1; From cd29f3bc6c3b2827ed6ba4f644ae84c8000ccfc2 Mon Sep 17 00:00:00 2001 From: Johannes M Dieterich Date: Wed, 12 Sep 2018 18:51:15 -0500 Subject: [PATCH 2/2] Fix copy paste error located by Jithun Nair. --- cmake/public/LoadHIP.cmake | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cmake/public/LoadHIP.cmake b/cmake/public/LoadHIP.cmake index ee943791c2ad93..842cf349cbef55 100644 --- a/cmake/public/LoadHIP.cmake +++ b/cmake/public/LoadHIP.cmake @@ -47,7 +47,7 @@ ENDIF() # ROCFFT_PATH IF(NOT DEFINED ENV{ROCFFT_PATH}) - SET(ROCBLAS_PATH ${ROCM_PATH}/rocfft) + SET(ROCFFT_PATH ${ROCM_PATH}/rocfft) ELSE() SET(ROCFFT_PATH $ENV{ROCFFT_PATH}) ENDIF()