AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-388958 - SWDEV-388929 - Kind can be hipMemcpyDefault for H2D and D2D and srcMemory can be allocated using hipHostMalloc for H2D.

Browse Source 
Change-Id: I187ff850ebb890b357ee03500105b71468508ea2
This commit is contained in:
Jaydeep Patel
2023-04-10 07:24:18 +00:00
committed by Jaydeepkumar Patel
parent 09a081235c
commit 7215451db9
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/src/hip_graph_internal.hpp
+3 -2
View File
@@ -1533,9 +1533,10 @@ class hipGraphMemcpyNodeToSymbol : public hipGraphMemcpyNode1D {
}
size_t dOffset = 0;
amd::Memory* srcMemory = getMemoryObject(src, dOffset);
if (srcMemory == nullptr && kind != hipMemcpyHostToDevice) {
if (srcMemory == nullptr && kind != hipMemcpyHostToDevice && kind != hipMemcpyDefault) {
return hipErrorInvalidValue;
} else if (srcMemory != nullptr && kind != hipMemcpyDeviceToDevice) {
} else if (srcMemory != nullptr && srcMemory->getMemFlags() == 0 &&
kind != hipMemcpyDeviceToDevice && kind != hipMemcpyDefault) {
return hipErrorInvalidValue;
} else if (kind == hipMemcpyHostToHost || kind == hipMemcpyDeviceToHost) {
return hipErrorInvalidValue;