MUDA
Loading...
Searching...
No Matches
compute_graph_node.inl
1#include <muda/compute_graph/compute_graph_var_manager.h>
2#include <muda/tools/launch_info_cache.h>
3namespace muda
4{
5template <typename NodeT, ComputeGraphNodeType Type>
6MUDA_INLINE ComputeGraphNode<NodeT, Type>::ComputeGraphNode(NodeId node_id, uint64_t access_graph_index)
7 : ComputeGraphNodeBase(enum_name(Type), node_id, access_graph_index, Type)
8{
9 if constexpr(Type == ComputeGraphNodeType::KernelNode)
10 {
11 auto n = std::string_view{
12 details::LaunchInfoCache::current_kernel_name().auto_select()};
13 if(n.empty() || n == "")
14 m_name += std::string(":~");
15 else
16 m_name += std::string(":") + std::string(n.data());
17 }
18}
19
20template <typename NodeT, ComputeGraphNodeType Type>
21MUDA_INLINE void ComputeGraphNode<NodeT, Type>::set_node(S<NodeT> node)
22{
23 m_node = node;
24 set_handle(m_node->handle());
25}
26} // namespace muda