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; 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()