Skip to content

Commit 4b8deaa

Browse files
authored
Merge pull request #103 from AMD-Lightning-Internal/amd/dev/rlieberm/CG-add-env-var
[OpenMP][AMDGPU] Add env var check for coarse grain memory pool map
2 parents 4c5817d + 3e6f45b commit 4b8deaa

File tree

1 file changed

+2
-2
lines changed
  • offload/plugins-nextgen/amdgpu/src

1 file changed

+2
-2
lines changed

offload/plugins-nextgen/amdgpu/src/rtl.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3498,8 +3498,8 @@ struct AMDGPUDeviceTy : public GenericDeviceTy, AMDGenericDeviceTy {
34983498
Error setCoarseGrainMemoryImpl(void *ptr, int64_t size,
34993499
bool set_attr = true) override final {
35003500
// If the table has not yet been created, check if the gpu arch is
3501-
// MI200 and create it.
3502-
if (!IsEquippedWithGFX90A)
3501+
// MI200 and create it, but only if USM Map is enabled.
3502+
if (!IsEquippedWithGFX90A || OMPX_DisableUsmMaps)
35033503
return Plugin::success();
35043504
if (!CoarseGrainMemoryTable)
35053505
CoarseGrainMemoryTable = new AMDGPUMemTypeBitFieldTable(

0 commit comments

Comments
 (0)