Skip to content

Commit

Permalink
Test err as warn for Advise API
Browse files Browse the repository at this point in the history
GeorgeWeb committed Nov 7, 2023
1 parent 89a55c7 commit b7664c0
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions source/adapters/cuda/enqueue.cpp
Original file line number Diff line number Diff line change
@@ -83,12 +83,8 @@ ur_result_t setCuMemAdvise(CUdeviceptr DevPtr, size_t Size,
CU_MEM_ADVISE_UNSET_READ_MOSTLY},
{UR_USM_ADVICE_FLAG_SET_PREFERRED_LOCATION,
CU_MEM_ADVISE_SET_PREFERRED_LOCATION},
{UR_USM_ADVICE_FLAG_CLEAR_PREFERRED_LOCATION,
CU_MEM_ADVISE_UNSET_PREFERRED_LOCATION},
{UR_USM_ADVICE_FLAG_SET_ACCESSED_BY_DEVICE,
CU_MEM_ADVISE_SET_ACCESSED_BY},
{UR_USM_ADVICE_FLAG_CLEAR_ACCESSED_BY_DEVICE,
CU_MEM_ADVISE_UNSET_ACCESSED_BY},
};
for (auto &FlagPair : URToCUMemAdviseDeviceFlagsMap) {
if (URAdviceFlags & FlagPair.first) {
@@ -117,7 +113,10 @@ ur_result_t setCuMemAdvise(CUdeviceptr DevPtr, size_t Size,
std::array<ur_usm_advice_flags_t, 4> UnmappedMemAdviceFlags = {
UR_USM_ADVICE_FLAG_SET_NON_ATOMIC_MOSTLY,
UR_USM_ADVICE_FLAG_CLEAR_NON_ATOMIC_MOSTLY,
UR_USM_ADVICE_FLAG_BIAS_CACHED, UR_USM_ADVICE_FLAG_BIAS_UNCACHED};
UR_USM_ADVICE_FLAG_BIAS_CACHED,
UR_USM_ADVICE_FLAG_BIAS_UNCACHED,
UR_USM_ADVICE_FLAG_CLEAR_ACCESSED_BY_DEVICE,
UR_USM_ADVICE_FLAG_CLEAR_PREFERRED_LOCATION};

for (auto &UnmappedFlag : UnmappedMemAdviceFlags) {
if (URAdviceFlags & UnmappedFlag) {

0 comments on commit b7664c0

Please sign in to comment.