SWDEV-315595 - Fix for hipGraphAddMemcpyNodeToSymbol -ve tests

Change-Id: If389f533567a236a598f16e4f2c477157beb2e56


[ROCm/clr commit: 15dd9fc1da]
这个提交包含在:
Sarbojit Sarkar 编写, Sarbojit Sarkar 于 2022-01-26 22:42:04 -05:00 提交
1 父节点 9b118a81d3
当前提交 6b86bb1618
1 个文件已更改
+7
+7
查看文件
@@ -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);
}