AI/rocm-systems
2
0
Vork 0
Code Kwesties Pull-aanvragen Actions Paketten Projecten Publicaties Wiki Activiteit

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

Bladeren bron 
Change-Id: Ice1df4d3722354f40c8b02ba35ed122f9b2e2f2d
This commit is contained in:
Ioannis Assiouras
2023-03-06 13:58:04 +00:00
bovenliggende 7715b3b398
commit 5837d83e46
1 gewijzigde bestanden met toevoegingen van 6 en 0 verwijderingen
Download Patch-bestand Download Diff-bestand Expand all files Collapse all files
hipamd/src/hip_platform.cpp
+6
Bestand weergeven
@@ -443,6 +443,9 @@ hipError_t hipModuleOccupancyMaxPotentialBlockSizeWithFlags(int* gridSize, int*
if ((gridSize == nullptr) || (blockSize == nullptr)) { if ((gridSize == nullptr) || (blockSize == nullptr)) {
HIP_RETURN(hipErrorInvalidValue); HIP_RETURN(hipErrorInvalidValue);
} }
if (flags != hipOccupancyDefault && flags != hipOccupancyDisableCachingOverride) {
HIP_RETURN(hipErrorInvalidValue);
}
const amd::Device& device = *hip::getCurrentDevice()->devices()[0]; const amd::Device& device = *hip::getCurrentDevice()->devices()[0];
int max_blocks_per_grid = 0; int max_blocks_per_grid = 0;
int num_blocks = 0; int num_blocks = 0;
@@ -484,6 +487,9 @@ hipError_t hipModuleOccupancyMaxActiveBlocksPerMultiprocessorWithFlags(
if (numBlocks == nullptr) { if (numBlocks == nullptr) {
HIP_RETURN(hipErrorInvalidValue); HIP_RETURN(hipErrorInvalidValue);
} }
if (flags != hipOccupancyDefault && flags != hipOccupancyDisableCachingOverride) {
HIP_RETURN(hipErrorInvalidValue);
}
const amd::Device& device = *hip::getCurrentDevice()->devices()[0]; const amd::Device& device = *hip::getCurrentDevice()->devices()[0];
int num_blocks = 0; int num_blocks = 0;