SWDEV-315595 - Fix for hipGraphAddMemcpyNodeToSymbol -ve tests

Change-Id: If389f533567a236a598f16e4f2c477157beb2e56


[ROCm/clr commit: 15dd9fc1da]
This commit is contained in:
Sarbojit Sarkar
2022-01-25 07:25:42 +00:00
committato da Sarbojit Sarkar
parent 9b118a81d3
commit 6b86bb1618
+7
Vedi File
@@ -1363,6 +1363,10 @@ hipError_t hipGraphAddMemcpyNodeToSymbol(hipGraphNode_t* pGraphNode, hipGraph_t
hipMemcpyKind kind) {
HIP_INIT_API(hipGraphAddMemcpyNodeToSymbol, pGraphNode, graph, pDependencies, numDependencies,
symbol, src, count, offset, kind);
if (pGraphNode == nullptr || graph == nullptr || src == nullptr ||
(pDependencies == nullptr && numDependencies > 0)) {
HIP_RETURN(hipErrorInvalidValue);
}
size_t sym_size = 0;
hipDeviceptr_t device_ptr = nullptr;
hipError_t status = ihipMemcpySymbol_validate(symbol, count, offset, sym_size, device_ptr);
@@ -1370,6 +1374,9 @@ hipError_t hipGraphAddMemcpyNodeToSymbol(hipGraphNode_t* pGraphNode, hipGraph_t
HIP_RETURN(status);
}
*pGraphNode = new hipGraphMemcpyNodeToSymbol(symbol, src, count, offset, kind);
if (*pGraphNode == nullptr) {
HIP_RETURN(hipErrorInvalidValue);
}
ihipGraphAddNode(*pGraphNode, graph, pDependencies, numDependencies);
HIP_RETURN(hipSuccess);
}