SWDEV-474159 - Query max LDS in Unit_hipDynamicShared

Change-Id: Ie78613ea1a87d69aff4f764368ea7a1893e021f5


[ROCm/hip-tests commit: 5e26ec0bb1]

projects/hip-tests/catch/include/hip_test_common.hh

@@ -237,27 +237,6 @@ static inline bool IsGfx11() {
 #endif
 }
 
-static inline bool IsMi350() {
-#if HT_NVIDIA
-  return false;
-#elif HT_AMD
-  int device = -1;
-  hipDeviceProp_t props{};
-  HIP_CHECK(hipGetDevice(&device));
-  HIP_CHECK(hipGetDeviceProperties(&props, device));
-  // Get GCN Arch Name and compare to check if it is gfx11
-  std::string arch = std::string(props.gcnArchName);
-  auto pos = arch.find("gfx950");
-  if (pos != std::string::npos)
-    return true;
-  else
-    return false;
-#else
-  std::cout << "Have to be either Nvidia or AMD platform, asserting" << std::endl;
-  assert(false);
-#endif
-}
-
 // Utility Functions
 namespace HipTest {
 static inline int getDeviceCount() {

projects/hip-tests/catch/unit/kernel/hipDynamicShared.cc

@@ -163,13 +163,10 @@ TEST_CASE("Unit_hipDynamicShared") {
   }
 
   SECTION("test case with float for max LDS size") {
-    if(IsMi350())
-    {
-      testExternShared<float>(1024, 160*1024/sizeof(float));
-    }else
-    {
-      testExternShared<float>(1024, 64*1024/sizeof(float));
-    }
+    int maxLDS = 0;
+    HIP_CHECK(hipDeviceGetAttribute(&maxLDS,
+                                  hipDeviceAttributeMaxSharedMemoryPerBlock, 0));
+    testExternShared<float>(1024, maxLDS/sizeof(float));
   }
 }