SWDEV-474159 - Query max LDS in Unit_hipDynamicShared

Change-Id: Ie78613ea1a87d69aff4f764368ea7a1893e021f5
2024-07-17 17:16:11 +01:00
parent 98ab0ef459
commit 5e26ec0bb1
2 changed files with 4 additions and 28 deletions
@@ -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));
  }
 }