From 5e466891916f4350ef2c5a26334c0ff8b854e595 Mon Sep 17 00:00:00 2001
From: Saleel Kudchadker <Saleel.Kudchadker@amd.com>
Date: Mon, 31 Jan 2022 16:17:06 -0800
Subject: [PATCH] SWDEV-301947 - Add new query for physical CU count

Add hipDeviceAttributePhysicalMultiprocessorCount to get all
available CUs for a device.

Change-Id: Ie6a2c6e958df9627f39c91d0651b89ef5bffd8fe


[ROCm/clr commit: 9a3b4a93d27e5c99dd12b411d07a5d0b88ad47d4]
---
 projects/clr/hipamd/src/hip_device_runtime.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/projects/clr/hipamd/src/hip_device_runtime.cpp b/projects/clr/hipamd/src/hip_device_runtime.cpp
index 642e40092f..ada0ff3617 100644
--- a/projects/clr/hipamd/src/hip_device_runtime.cpp
+++ b/projects/clr/hipamd/src/hip_device_runtime.cpp
@@ -319,6 +319,9 @@ hipError_t hipDeviceGetAttribute(int* pi, hipDeviceAttribute_t attr, int device)
   case hipDeviceAttributeImageSupport:
     *pi = static_cast<int>(g_devices[device]->devices()[0]->info().imageSupport_);
     break;
+  case hipDeviceAttributePhysicalMultiProcessorCount:
+    *pi = g_devices[device]->devices()[0]->info().maxPhysicalComputeUnits_;
+    break;
   default:
     HIP_RETURN(hipErrorInvalidValue);
   }