Make hipHostMalloc() respect hipSetDevice()

Change-Id: Ibdb666fe8dd049735df2288878501a66f7eedc28
This commit is contained in:
Tao Sang
2020-06-05 15:21:59 -04:00
کامیت شده توسط Tao Sang
والد e4177b75bc
کامیت db10d42e50
4فایلهای تغییر یافته به همراه30 افزوده شده و 12 حذف شده
+5 -4
مشاهده پرونده
@@ -1454,12 +1454,13 @@ bool SvmBuffer::Contains(uintptr_t ptr) {
}
// The allocation flags are ignored for now.
void* SvmBuffer::malloc(Context& context, cl_svm_mem_flags flags, size_t size, size_t alignment) {
void* SvmBuffer::malloc(Context& context, cl_svm_mem_flags flags, size_t size, size_t alignment,
const amd::Device* curDev) {
bool atomics = (flags & CL_MEM_SVM_ATOMICS) != 0;
void* ret = context.svmAlloc(size, alignment, flags);
if (ret == NULL) {
void* ret = context.svmAlloc(size, alignment, flags, curDev);
if (ret == nullptr) {
LogError("Unable to allocate aligned memory");
return NULL;
return nullptr;
}
uintptr_t ret_u = reinterpret_cast<uintptr_t>(ret);
Add(ret_u, ret_u + size);