From d3152b14153065a8e2961cd9e165fa323cb6f82e Mon Sep 17 00:00:00 2001
From: ROCm CI Service Account <66695075+rocm-ci@users.noreply.github.com>
Date: Mon, 6 Mar 2023 21:11:25 +0530
Subject: [PATCH] SWDEV-373630 - Modified occupancy query tests to conform with
 new error messages. (#3175)

Change-Id: I20e16e78cb5f170609b33e0770b7fca5bff2b280

[ROCm/hip commit: 5557d17afc6abba779234aaf9498e79d923f04f2]
---
 .../runtimeApi/occupancy/hipOccupancyMaxPotentialBlockSize.cpp  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/hip/tests/src/runtimeApi/occupancy/hipOccupancyMaxPotentialBlockSize.cpp b/projects/hip/tests/src/runtimeApi/occupancy/hipOccupancyMaxPotentialBlockSize.cpp
index e06ea9c600..49f0ed1bf7 100644
--- a/projects/hip/tests/src/runtimeApi/occupancy/hipOccupancyMaxPotentialBlockSize.cpp
+++ b/projects/hip/tests/src/runtimeApi/occupancy/hipOccupancyMaxPotentialBlockSize.cpp
@@ -82,7 +82,7 @@ bool argValidation() {
 #ifndef __HIP_PLATFORM_NVIDIA__
   // nvcc doesnt support kernelfunc(NULL) for api
   ret = hipOccupancyMaxPotentialBlockSize(&gridSize, &blockSize, NULL, 0, 0);
-  if (ret != hipErrorInvalidValue) {
+  if (ret != hipErrorInvalidValue && ret != hipErrorInvalidDeviceFunction) {
     printf("ArgValidation : Inappropritate error value returned for"
            " kernelfunc(NULL). gridSize %d, blkSize %d, Error: '%s'(%d)\n",
            gridSize, blockSize, hipGetErrorString(ret), ret);