SWDEV-399734, SWDEV-399733 - update maxGridDim for y and z
Change-Id: Id5b1dd2491798dd20a0672362f445ab4de164f6d
Этот коммит содержится в:
коммит произвёл
Payam Ghafari
родитель
a4c5c44008
Коммит
fb82b292c2
@@ -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();
|
||||
|
||||
@@ -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;
|
||||
|
||||
Ссылка в новой задаче
Block a user