SWDEV-373630 - Added checks for invalid flags in hipOccupancy query functions

Change-Id: Ice1df4d3722354f40c8b02ba35ed122f9b2e2f2d
Этот коммит содержится в:
Ioannis Assiouras
2023-03-06 13:58:04 +00:00
родитель 7715b3b398
Коммит 5837d83e46
+6
Просмотреть файл
@@ -443,6 +443,9 @@ hipError_t hipModuleOccupancyMaxPotentialBlockSizeWithFlags(int* gridSize, int*
if ((gridSize == nullptr) || (blockSize == nullptr)) {
HIP_RETURN(hipErrorInvalidValue);
}
if (flags != hipOccupancyDefault && flags != hipOccupancyDisableCachingOverride) {
HIP_RETURN(hipErrorInvalidValue);
}
const amd::Device& device = *hip::getCurrentDevice()->devices()[0];
int max_blocks_per_grid = 0;
int num_blocks = 0;
@@ -484,6 +487,9 @@ hipError_t hipModuleOccupancyMaxActiveBlocksPerMultiprocessorWithFlags(
if (numBlocks == nullptr) {
HIP_RETURN(hipErrorInvalidValue);
}
if (flags != hipOccupancyDefault && flags != hipOccupancyDisableCachingOverride) {
HIP_RETURN(hipErrorInvalidValue);
}
const amd::Device& device = *hip::getCurrentDevice()->devices()[0];
int num_blocks = 0;