diff --git a/aten/src/THC/generic/THCTensorIndex.cu b/aten/src/THC/generic/THCTensorIndex.cu index 0b985487bbb14f..c6204ff704a39b 100644 --- a/aten/src/THC/generic/THCTensorIndex.cu +++ b/aten/src/THC/generic/THCTensorIndex.cu @@ -560,7 +560,8 @@ void THCTensor_(indexSelect)(THCState *state, THCTensor *dst, THCTensor *src, in indexSelectSmallIndex \ <<>>( \ dstInfo, srcInfo, indicesInfo, \ - dstSelectDim, srcSelectDim, sliceSize, srcSelectDimSize); + dstSelectDim, srcSelectDim, static_cast(sliceSize), \ + srcSelectDimSize); #define LARGE_INDEX(TENSOR_TYPE, TYPE, \ DST_DIM, SRC_DIM, IDX_DIM, IDX_IS_MAJOR) \ @@ -568,8 +569,8 @@ void THCTensor_(indexSelect)(THCState *state, THCTensor *dst, THCTensor *src, in DST_DIM, SRC_DIM, IDX_DIM, IDX_IS_MAJOR> \ <<>>( \ dstInfo, srcInfo, indicesInfo, \ - dstSelectDim, srcSelectDim, dstTotalSize, \ - (IDX_IS_MAJOR) ? sliceSize : numIndices, \ + dstSelectDim, srcSelectDim, static_cast(dstTotalSize), \ + static_cast((IDX_IS_MAJOR) ? sliceSize : numIndices), \ srcSelectDimSize); dim3 smallIndexGrid(std::min(THCCeilDiv(sliceSize, (ptrdiff_t)128), (ptrdiff_t)(mpc * 8))); diff --git a/aten/src/THCUNN/LookupTableBag.cu b/aten/src/THCUNN/LookupTableBag.cu index c579b93bcb2331..2c9e09cdb7624d 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()