AI/rocm-systems
2
0
Förgrening 0
Kod Ärenden Pull-förfrågningar Actions Packages Projekt Släpp Wiki Aktiviteter

SWDEV-240806 - Added Implementation for few more graph APIs

Bläddra i källkod 
Change-Id: I1e443cf1b4e3a1e85f6fb9c1db8fdfa7c1fdfe06
This commit is contained in:
Anusha GodavarthySurya
2021-07-07 03:52:11 -07:00
committad av Anusha Godavarthy Surya
förälder b77fe6cc9f
incheckning 6e7ea5b612
3 ändrade filer med 35 tillägg och 39 borttagningar
Download Patch File Download Diff File Expand all files Collapse all files
include/hip/hip_runtime_api.h
+26
Visa fil
@@ -4160,6 +4160,32 @@ hipError_t hipGraphAddMemcpyNode(hipGraphNode_t* pGraphNode, hipGraph_t graph,
hipError_t hipGraphAddMemsetNode(hipGraphNode_t* pGraphNode, hipGraph_t graph,
const hipGraphNode_t* pDependencies, size_t numDependencies,
const hipMemsetParams* pMemsetParams);
// Returns graph nodes
hipError_t hipGraphGetNodes(hipGraph_t graph, hipGraphNode_t* nodes, size_t* numNodes);
// Returns graph's root nodes.
hipError_t hipGraphGetRootNodes(hipGraph_t graph, hipGraphNode_t* pRootNodes,
size_t* pNumRootNodes);
// Returns a kernel node's parameters.
hipError_t hipGraphKernelNodeGetParams(hipGraphNode_t node, hipKernelNodeParams* pNodeParams);
// Sets a kernel node's parameters.
hipError_t hipGraphKernelNodeSetParams(hipGraphNode_t node, const hipKernelNodeParams* pNodeParams);
// Returns a memcpy node's parameters.
hipError_t hipGraphMemcpyNodeGetParams(hipGraphNode_t node, hipMemcpy3DParms* pNodeParams);
// Sets a memcpy node's parameters.
hipError_t hipGraphMemcpyNodeSetParams(hipGraphNode_t node, const hipMemcpy3DParms* pNodeParams);
// Returns a memset node's parameters.
hipError_t hipGraphMemsetNodeGetParams(hipGraphNode_t node, hipMemsetParams* pNodeParams);
// Sets a memset node's parameters.
hipError_t hipGraphMemsetNodeSetParams(hipGraphNode_t node, const hipMemsetParams* pNodeParams);
// Sets the parameters for a kernel node in the given graphExec.
hipError_t hipGraphExecKernelNodeSetParams(hipGraphExec_t hGraphExec, hipGraphNode_t node,
const hipKernelNodeParams* pNodeParams);
// Adds dependency edges to a graph.
hipError_t hipGraphAddDependencies(hipGraph_t graph, const hipGraphNode_t* from,
const hipGraphNode_t* to, size_t numDependencies);
// Creates an empty node and adds it to a graph.
hipError_t hipGraphAddEmptyNode(hipGraphNode_t* pGraphNode, hipGraph_t graph,
const hipGraphNode_t* pDependencies, size_t numDependencies);
#endif
// doxygen end graph API
/**