diff --git a/projects/clr/hipamd/src/hip_graph_internal.hpp b/projects/clr/hipamd/src/hip_graph_internal.hpp index 76e9adc7df..d3cac3a09f 100644 --- a/projects/clr/hipamd/src/hip_graph_internal.hpp +++ b/projects/clr/hipamd/src/hip_graph_internal.hpp @@ -1344,6 +1344,12 @@ class hipGraphMemcpyNode1D : public hipGraphNode { memcpyDirection = "DtoH"; } else if ((srcMemory != nullptr) && (dstMemory != nullptr)) { memcpyDirection = "DtoD"; + } else { + if (kind_ == hipMemcpyHostToDevice) { + memcpyDirection = "HtoD"; + } else if (kind_ == hipMemcpyDeviceToHost) { + memcpyDirection = "DtoH"; + } } std::string label; if (flag == hipGraphDebugDotFlagsMemcpyNodeParams || flag == hipGraphDebugDotFlagsVerbose) {