SWDEV-231145 - Add a validation for uniform launch
Change-Id: I68d61690c50060d9dcca117b248dddf69204a23f
이 커밋은 다음에 포함됨:
1 부모
73698479a9
커밋
e82eecf33e
3 파일이 변경됨
+21
-7
검색과 일치하는 파일 없음
@@ -375,6 +375,15 @@ hipError_t ihipModuleLaunchKernel(hipFunction_t f, uint32_t globalWorkSizeX,
|
||||
if (globalWorkSizeY < blockDimY) blockDimY = globalWorkSizeY;
|
||||
if (globalWorkSizeZ < blockDimZ) blockDimZ = globalWorkSizeZ;
|
||||
|
||||
auto device = g_devices[deviceId]->devices()[0];
|
||||
// Check if it's a uniform kernel and validate dimensions
|
||||
if (kernel->getDeviceKernel(*device)->getUniformWorkGroupSize()) {
|
||||
if (((globalWorkSizeX % blockDimX) != 0) ||
|
||||
((globalWorkSizeY % blockDimY) != 0) ||
|
||||
((globalWorkSizeZ % blockDimZ) != 0)) {
|
||||
return hipErrorInvalidValue;
|
||||
}
|
||||
}
|
||||
amd::Command* command = nullptr;
|
||||
hip::Stream* hip_stream = hip::getStream(hStream);
|
||||
status = ihipLaunchKernelCommand(command, f, globalWorkSizeX, globalWorkSizeY, globalWorkSizeZ,
|
||||
|
||||
새 이슈에서 참조
사용자 차단