AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-399734, SWDEV-399733 - update maxGridDim for y and z

Browse Source 
Change-Id: Id5b1dd2491798dd20a0672362f445ab4de164f6d


[ROCm/clr commit: fb82b292c2]
This commit is contained in:
pghafari
2023-05-16 18:12:46 -04:00
committed by Payam Ghafari
parent 2e664d2492
commit 9eda493469
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/clr/hipamd/src/hip_device.cpp
+1 -1
View File
@@ -286,7 +286,7 @@ hipError_t ihipGetDeviceProperties(hipDeviceProp_t* props, hipDevice_t device) {
auto* deviceHandle = g_devices[device]->devices()[0];
constexpr auto int32_max = static_cast<uint64_t>(std::numeric_limits<int32_t>::max());
constexpr auto uint16_max = static_cast<uint64_t>(std::numeric_limits<uint16_t>::max());
constexpr auto uint16_max = static_cast<uint64_t>(std::numeric_limits<uint16_t>::max())+1;
hipDeviceProp_t deviceProps = {0};
const auto& info = deviceHandle->info();
projects/clr/hipamd/src/hip_module.cpp
+2 -2
View File
@@ -440,8 +440,8 @@ hipError_t hipModuleLaunchKernel(hipFunction_t f, uint32_t gridDimX, uint32_t gr
STREAM_CAPTURE(hipModuleLaunchKernel, hStream, f, gridDimX, gridDimY, gridDimZ, blockDimX,
blockDimY, blockDimZ, sharedMemBytes, kernelParams, extra);
if (gridDimX > std::numeric_limits<int32_t>::max() ||
gridDimY > std::numeric_limits<uint16_t>::max() ||
gridDimZ > std::numeric_limits<uint16_t>::max()) {
gridDimY > std::numeric_limits<uint16_t>::max()+1 ||
gridDimZ > std::numeric_limits<uint16_t>::max()+1) {
HIP_RETURN(hipErrorInvalidValue);
}
size_t globalWorkSizeX = static_cast<size_t>(gridDimX) * blockDimX;