AI/rocm-systems
2
0
Forkuj 0
Kod Zgłoszenia Oczekujące zmiany Actions Packages Projekty Wydania Wiki Aktywność

SWDEV-421021 - Add APIs cuMemcpyNodeGet/Set params

Przeglądaj źródła 
Signed-off-by: sdashmiz <shadi.dashmiz@amd.com>
Change-Id: I24bc0da56aad34c9d5876a3d83b59515f11dc3ea
This commit is contained in:
sdashmiz
2024-01-18 10:52:32 -05:00
zatwierdzone przez Shadi Dashmiz
rodzic ff96b23dda
commit 57e79802cd
6 zmienionych plików z 48 dodań i 4 usunięć
Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files
hipamd/src/hip_graph.cpp
+19
Wyświetl plik
@@ -3271,4 +3271,23 @@ hipError_t hipGraphExecNodeSetParams(hipGraphExec_t graphExec, hipGraphNode_t no
}
HIP_RETURN(ihipGraphNodeSetParams(clonedNode, nodeParams));
}
hipError_t hipDrvGraphMemcpyNodeGetParams(hipGraphNode_t hNode, HIP_MEMCPY3D* nodeParams) {
HIP_INIT_API(hipDrvGraphMemcpyNodeGetParams, hNode, nodeParams);
if (!hip::GraphNode::isNodeValid(reinterpret_cast<hip::GraphNode*>(hNode)) ||
nodeParams == nullptr) {
HIP_RETURN(hipErrorInvalidValue);
}
reinterpret_cast<hip::GraphDrvMemcpyNode*>(hNode)->GetParams(nodeParams);
HIP_RETURN(hipSuccess);
}
hipError_t hipDrvGraphMemcpyNodeSetParams(hipGraphNode_t hNode, const HIP_MEMCPY3D* nodeParams) {
HIP_INIT_API(hipDrvGraphMemcpyNodeSetParams, hNode, nodeParams);
if (!hip::GraphNode::isNodeValid(reinterpret_cast<hip::GraphNode*>(hNode)) ||
nodeParams == nullptr) {
HIP_RETURN(hipErrorInvalidValue);
}
HIP_RETURN(reinterpret_cast<hip::GraphDrvMemcpyNode*>(hNode)->SetParams(nodeParams));
}
} // namespace hip