Skip to content

Commit

Permalink
Merge pull request openucx#12 from edgargabriel/topic/memtype-detecti…
Browse files Browse the repository at this point in the history
…on-fix2

UCM/ROCM: delay memtype detection to md's
  • Loading branch information
edgargabriel authored Jun 13, 2022
2 parents 891cf48 + 9453b2c commit f363042
Showing 1 changed file with 1 addition and 10 deletions.
11 changes: 1 addition & 10 deletions src/ucm/rocm/rocmmem.c
Original file line number Diff line number Diff line change
Expand Up @@ -111,18 +111,9 @@ hsa_status_t ucm_hsa_amd_memory_pool_allocate(
hsa_amd_memory_pool_t memory_pool, size_t size,
uint32_t flags, void** ptr)
{
ucs_memory_type_t type = UCS_MEMORY_TYPE_ROCM;
uint32_t pool_flags = 0;
ucs_memory_type_t type = UCS_MEMORY_TYPE_LAST;
hsa_status_t status;

status = hsa_amd_memory_pool_get_info(memory_pool,
HSA_AMD_MEMORY_POOL_INFO_GLOBAL_FLAGS,
&pool_flags);
if (status == HSA_STATUS_SUCCESS &&
!(pool_flags & HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_COARSE_GRAINED)) {
type = UCS_MEMORY_TYPE_ROCM_MANAGED;
}

ucm_event_enter();

status = ucm_orig_hsa_amd_memory_pool_allocate(memory_pool, size, flags, ptr);
Expand Down

0 comments on commit f363042

Please sign in to comment.