Coverage for cuda / core / graph / _graph_definition.pyx: 93.22%
118 statements
« prev ^ index » next coverage.py v7.14.0, created at 2026-05-22 01:37 +0000
« prev ^ index » next coverage.py v7.14.0, created at 2026-05-22 01:37 +0000
1# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2#
3# SPDX-License-Identifier: Apache-2.0
5"""GraphDefinition: explicit CUDA graph definition."""
7from __future__ import annotations
9from libc.stddef cimport size_t
11from libcpp.vector cimport vector
13from cuda.bindings cimport cydriver
15from cuda.core.graph._graph_node cimport GraphNode
16from cuda.core._resource_handles cimport (
17 GraphHandle,
18 as_cu,
19 as_intptr,
20 as_py,
21 create_graph_handle,
22 create_graph_node_handle,
23)
24from cuda.core._utils.cuda_utils cimport HANDLE_RETURN
26from cuda.core._utils.cuda_utils import driver
28from cuda.core.typing import GraphMemoryType
30__all__ = ['GraphCondition', 'GraphDefinition']
33cdef class GraphCondition:
34 """A condition variable for conditional graph nodes.
36 Created by :meth:`GraphDefinition.create_condition` (or
37 :meth:`GraphBuilder.create_condition`) and passed to
38 conditional-node builder methods (:meth:`~GraphDefinition.if_then`,
39 :meth:`~GraphDefinition.if_else`, :meth:`~GraphDefinition.while_loop`,
40 :meth:`~GraphDefinition.switch`). The underlying value is set at
41 runtime by device code via ``cudaGraphSetConditional``.
43 A :class:`GraphCondition` may be passed directly as a kernel
44 argument to ``launch()``: the launcher unwraps it to the underlying
45 ``CUgraphConditionalHandle`` value so device code can update the
46 condition.
47 """
49 @staticmethod
50 cdef GraphCondition _from_handle(cydriver.CUgraphConditionalHandle c_handle):
51 cdef GraphCondition self = GraphCondition.__new__(GraphCondition) 2WlXlYlZl0l1l2l3l4l5l6l7l8l9l!l#l$l%l'l(l)l*l+l,lq MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e -l.lxd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
52 self._c_handle = c_handle 2WlXlYlZl0l1l2l3l4l5l6l7l8l9l!l#l$l%l'l(l)l*l+l,lq MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e -l.lxd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
53 return self 2WlXlYlZl0l1l2l3l4l5l6l7l8l9l!l#l$l%l'l(l)l*l+l,lq MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e -l.lxd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
55 def __repr__(self) -> str:
56 return f"<GraphCondition handle=0x{<unsigned long long>self._c_handle:x}>" 2Yd
58 def __eq__(self, other) -> bool:
59 if not isinstance(other, GraphCondition): 24m5m6m7m9d!dJe#dxdDdgdzcAchdidjdkdI J ldBcmdudK L EdFdGdzdHdIdJdKdLdAdBd
60 return NotImplemented 2xdDdgdzcAchdidjdkdI J ldBcmdudK L EdFdGdzdHdIdJdKd
61 return self._c_handle == (<GraphCondition>other)._c_handle 24m5m6m7m9d!dJe#dxdLdAdBd
63 def __hash__(self) -> int:
64 return hash(<unsigned long long>self._c_handle) 2OdPdndddedodpdqdrd3 4 sdfdtdyd5 6 QdRdSdCdTdUdVdWdXdZd0dAdBd
66 @property
67 def handle(self) -> driver.CUgraphConditionalHandle:
68 """The raw CUgraphConditionalHandle as an int."""
69 return <unsigned long long>self._c_handle 2WlXlYlZl0l1l2l3l4l5l6l7l8l9l!l#l$l%l'l(l)l*l+l,l-l.lxd
72cdef class GraphDefinition:
73 """A graph definition.
75 A GraphDefinition is used to construct a graph explicitly by adding nodes
76 and specifying dependencies. Once construction is complete, call
77 instantiate() to obtain an executable Graph.
78 """
80 def __init__(self):
81 """Create a new empty graph definition."""
82 cdef cydriver.CUgraph graph = NULL 2/l:l;l=l?l@l[l]l^lHgIgJg_lKeKgLg`l{l|l}l~lY ambmcmdm6demfmgmhmimjmkmlmmmnmompmqmrmsmtmumvmwmxmymzmAmBmCmDmEmFmGmHmImMgNgLeMeOgPgQgRgSgMbNbTgNeUgVgWgXgObPbYgZgOePe0g1g2g3g4gQbRb5gQe6g7g8g9gSbTb!g#g$gJmRe%g'g(g)g*gSe+g,g-g.gTeUe/g:g;g=g?gUbVb@gVe[g]g^g_gWbXb`g{gWeXe|g}g~gahbhYbZbchYedhehfhgh0b1bhhihZe0ejhkhlhmhnh2b3boh1ephqhrhsh4b5bthuh2e3evhwhxhyhzh6b7bAh4eBhChDhEh8b9bFhGhHh5eIhJhKhLhMh6eNhOhPhQhRhSh7eThUhvdKmLmMmNm3dOm4dVhWhXh$dwd%df g h i a b c d 'd(dMdYhZh0h8e)d1h2h*d9e3h+d!e,dNd-de j k l m n 4h5h6h7ho 8hr 9h7d!h#exd$e#h$h%h'h(h!b#b)h%e*hkk$b%b+h'eOd(e,h-h.hAl/h'b(b:h)e;hlk)b*b{d*e+e=h,eZ 0 -e.e?h1 2 @h/ePd:e[h]h^hBl_h+b,b`h;e{hmk-b.b|d}d~ds t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 |h=eQd?e}h~haiClbi/b:bci@edink;b=bei[eRd]efigihiDlii?b@bji^ekiok[b]b_e.daebe`ecew x de{e|ey z li}eSd~eminioiElpi^b_bqiafripk`b{bbfcf7 8 9 ! df# $ % ' eeefffgf( ) hfsi* + iffeCdgejfkflfmf, - nfheofti. / : ; = ? @ [ ] ^ _ uipfTdqfviwixiFlyi|b}bzirfAiqk~bacsftfufvfwf` { iexf| } BiyfUdzfCiDiEiGlFibcccGiAfHirkdcec/djekeleA B C D meBfCfDf~ abEfbbcbIiFfVdGfJiKiLiHlMifcgcNiHfOiskhcicPiIfJfQiRiSiTijckcUiKfVilcmcWiLfWdMfXiYiZiIl0incoc1iNf2itkpcqcdb3iOfXdPf4i5i6iJl7ircsc8iQf9iuktcucneRfSf!iTfebfbUfVf#igbhb$iWfDdXf%i'i(iql)ivcwc*iYf+ivkxcycoepeqeE F G H gdzcAchdidjdkdI J ldBcmdudK L ,iZfEd0f-i.i/irl:iCcDc;i1f=iwkEcFc?i2fFd3f@i[i]isl^iGcHc_i4f`ixkIcJc5f:drese6fteM N ue7f8fO P {i9fGd!f|i}i~itlajKcLcbj#fcjykMcNc$f%fibjbkblb'fmbnbobpbve(f)f*fqbrb+fdjsbtbwe;dzd=dxeyezeAeQ R Be?dCe,fS T ubvbwbxbybzbAbBbCbej-fHd.ffjgjhjulijOcPcjj/fkjzkQcRc:f;f=f?f@fDbEbDe[fFbGblj]fId^fmjnjojvlpjScTcqj_frjAkUcVc@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsj~fJdagtjujvjwlwjWcXcxjbgyjBkYcZczjcgdgAjBjCjDj0c1cEjegFj2c3cGjfgKdggHjIjJjxlKj4c5cLjhgMjCk6c7cLbNjigYdjgOjPjQjKlRj8c9cSjkgTjDk!c#cUjlgLdmgVjWjXjylYj$c%cZjng0jEk'c(c1jogZdpg2j3j4jLl5j)c*c6jqg7jFk+c,c8jrg0dsg9j!j#jMl$j-c.c%jtg'jGk/c:c(jugAdvg)j*j+jHk,j;c=c-jwg.jIk?c@c/jxgBdyg:j;j=jJk?j[c]c@jzg[jKk^c_c]jAg1dBg^j_j`jPm{j`c{c|jCg}jLk|c}c~jDg2dEgakbkckQmdk~cadekFgfkMkbdcd
83 with nogil: 2/l:l;l=l?l@l[l]l^lHgIgJg_lKeKgLg`l{l|l}l~lY ambmcmdm6demfmgmhmimjmkmlmmmnmompmqmrmsmtmumvmwmxmymzmAmBmCmDmEmFmGmHmImMgNgLeMeOgPgQgRgSgMbNbTgNeUgVgWgXgObPbYgZgOePe0g1g2g3g4gQbRb5gQe6g7g8g9gSbTb!g#g$gJmRe%g'g(g)g*gSe+g,g-g.gTeUe/g:g;g=g?gUbVb@gVe[g]g^g_gWbXb`g{gWeXe|g}g~gahbhYbZbchYedhehfhgh0b1bhhihZe0ejhkhlhmhnh2b3boh1ephqhrhsh4b5bthuh2e3evhwhxhyhzh6b7bAh4eBhChDhEh8b9bFhGhHh5eIhJhKhLhMh6eNhOhPhQhRhSh7eThUhvdKmLmMmNm3dOm4dVhWhXh$dwd%df g h i a b c d 'd(dMdYhZh0h8e)d1h2h*d9e3h+d!e,dNd-de j k l m n 4h5h6h7ho 8hr 9h7d!h#exd$e#h$h%h'h(h!b#b)h%e*hkk$b%b+h'eOd(e,h-h.hAl/h'b(b:h)e;hlk)b*b{d*e+e=h,eZ 0 -e.e?h1 2 @h/ePd:e[h]h^hBl_h+b,b`h;e{hmk-b.b|d}d~ds t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 |h=eQd?e}h~haiClbi/b:bci@edink;b=bei[eRd]efigihiDlii?b@bji^ekiok[b]b_e.daebe`ecew x de{e|ey z li}eSd~eminioiElpi^b_bqiafripk`b{bbfcf7 8 9 ! df# $ % ' eeefffgf( ) hfsi* + iffeCdgejfkflfmf, - nfheofti. / : ; = ? @ [ ] ^ _ uipfTdqfviwixiFlyi|b}bzirfAiqk~bacsftfufvfwf` { iexf| } BiyfUdzfCiDiEiGlFibcccGiAfHirkdcec/djekeleA B C D meBfCfDf~ abEfbbcbIiFfVdGfJiKiLiHlMifcgcNiHfOiskhcicPiIfJfQiRiSiTijckcUiKfVilcmcWiLfWdMfXiYiZiIl0incoc1iNf2itkpcqcdb3iOfXdPf4i5i6iJl7ircsc8iQf9iuktcucneRfSf!iTfebfbUfVf#igbhb$iWfDdXf%i'i(iql)ivcwc*iYf+ivkxcycoepeqeE F G H gdzcAchdidjdkdI J ldBcmdudK L ,iZfEd0f-i.i/irl:iCcDc;i1f=iwkEcFc?i2fFd3f@i[i]isl^iGcHc_i4f`ixkIcJc5f:drese6fteM N ue7f8fO P {i9fGd!f|i}i~itlajKcLcbj#fcjykMcNc$f%fibjbkblb'fmbnbobpbve(f)f*fqbrb+fdjsbtbwe;dzd=dxeyezeAeQ R Be?dCe,fS T ubvbwbxbybzbAbBbCbej-fHd.ffjgjhjulijOcPcjj/fkjzkQcRc:f;f=f?f@fDbEbDe[fFbGblj]fId^fmjnjojvlpjScTcqj_frjAkUcVc@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsj~fJdagtjujvjwlwjWcXcxjbgyjBkYcZczjcgdgAjBjCjDj0c1cEjegFj2c3cGjfgKdggHjIjJjxlKj4c5cLjhgMjCk6c7cLbNjigYdjgOjPjQjKlRj8c9cSjkgTjDk!c#cUjlgLdmgVjWjXjylYj$c%cZjng0jEk'c(c1jogZdpg2j3j4jLl5j)c*c6jqg7jFk+c,c8jrg0dsg9j!j#jMl$j-c.c%jtg'jGk/c:c(jugAdvg)j*j+jHk,j;c=c-jwg.jIk?c@c/jxgBdyg:j;j=jJk?j[c]c@jzg[jKk^c_c]jAg1dBg^j_j`jPm{j`c{c|jCg}jLk|c}c~jDg2dEgakbkckQmdk~cadekFgfkMkbdcd
84 HANDLE_RETURN(cydriver.cuGraphCreate(&graph, 0)) 2/l:l;l=l?l@l[l]l^lHgIgJg_lKeKgLg`l{l|l}l~lY ambmcmdm6demfmgmhmimjmkmlmmmnmompmqmrmsmtmumvmwmxmymzmAmBmCmDmEmFmGmHmImMgNgLeMeOgPgQgRgSgMbNbTgNeUgVgWgXgObPbYgZgOePe0g1g2g3g4gQbRb5gQe6g7g8g9gSbTb!g#g$gJmRe%g'g(g)g*gSe+g,g-g.gTeUe/g:g;g=g?gUbVb@gVe[g]g^g_gWbXb`g{gWeXe|g}g~gahbhYbZbchYedhehfhgh0b1bhhihZe0ejhkhlhmhnh2b3boh1ephqhrhsh4b5bthuh2e3evhwhxhyhzh6b7bAh4eBhChDhEh8b9bFhGhHh5eIhJhKhLhMh6eNhOhPhQhRhSh7eThUhvdKmLmMmNm3dOm4dVhWhXh$dwd%df g h i a b c d 'd(dMdYhZh0h8e)d1h2h*d9e3h+d!e,dNd-de j k l m n 4h5h6h7ho 8hr 9h7d!h#exd$e#h$h%h'h(h!b#b)h%e*hkk$b%b+h'eOd(e,h-h.hAl/h'b(b:h)e;hlk)b*b{d*e+e=h,eZ 0 -e.e?h1 2 @h/ePd:e[h]h^hBl_h+b,b`h;e{hmk-b.b|d}d~ds t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 |h=eQd?e}h~haiClbi/b:bci@edink;b=bei[eRd]efigihiDlii?b@bji^ekiok[b]b_e.daebe`ecew x de{e|ey z li}eSd~eminioiElpi^b_bqiafripk`b{bbfcf7 8 9 ! df# $ % ' eeefffgf( ) hfsi* + iffeCdgejfkflfmf, - nfheofti. / : ; = ? @ [ ] ^ _ uipfTdqfviwixiFlyi|b}bzirfAiqk~bacsftfufvfwf` { iexf| } BiyfUdzfCiDiEiGlFibcccGiAfHirkdcec/djekeleA B C D meBfCfDf~ abEfbbcbIiFfVdGfJiKiLiHlMifcgcNiHfOiskhcicPiIfJfQiRiSiTijckcUiKfVilcmcWiLfWdMfXiYiZiIl0incoc1iNf2itkpcqcdb3iOfXdPf4i5i6iJl7ircsc8iQf9iuktcucneRfSf!iTfebfbUfVf#igbhb$iWfDdXf%i'i(iql)ivcwc*iYf+ivkxcycoepeqeE F G H gdzcAchdidjdkdI J ldBcmdudK L ,iZfEd0f-i.i/irl:iCcDc;i1f=iwkEcFc?i2fFd3f@i[i]isl^iGcHc_i4f`ixkIcJc5f:drese6fteM N ue7f8fO P {i9fGd!f|i}i~itlajKcLcbj#fcjykMcNc$f%fibjbkblb'fmbnbobpbve(f)f*fqbrb+fdjsbtbwe;dzd=dxeyezeAeQ R Be?dCe,fS T ubvbwbxbybzbAbBbCbej-fHd.ffjgjhjulijOcPcjj/fkjzkQcRc:f;f=f?f@fDbEbDe[fFbGblj]fId^fmjnjojvlpjScTcqj_frjAkUcVc@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsj~fJdagtjujvjwlwjWcXcxjbgyjBkYcZczjcgdgAjBjCjDj0c1cEjegFj2c3cGjfgKdggHjIjJjxlKj4c5cLjhgMjCk6c7cLbNjigYdjgOjPjQjKlRj8c9cSjkgTjDk!c#cUjlgLdmgVjWjXjylYj$c%cZjng0jEk'c(c1jogZdpg2j3j4jLl5j)c*c6jqg7jFk+c,c8jrg0dsg9j!j#jMl$j-c.c%jtg'jGk/c:c(jugAdvg)j*j+jHk,j;c=c-jwg.jIk?c@c/jxgBdyg:j;j=jJk?j[c]c@jzg[jKk^c_c]jAg1dBg^j_j`jPm{j`c{c|jCg}jLk|c}c~jDg2dEgakbkckQmdk~cadekFgfkMkbdcd
85 self._h_graph = create_graph_handle(graph) 2/l:l;l=l?l@l[l]l^lHgIgJg_lKeKgLg`l{l|l}l~lY ambmcmdm6demfmgmhmimjmkmlmmmnmompmqmrmsmtmumvmwmxmymzmAmBmCmDmEmFmGmHmImMgNgLeMeOgPgQgRgSgMbNbTgNeUgVgWgXgObPbYgZgOePe0g1g2g3g4gQbRb5gQe6g7g8g9gSbTb!g#g$gJmRe%g'g(g)g*gSe+g,g-g.gTeUe/g:g;g=g?gUbVb@gVe[g]g^g_gWbXb`g{gWeXe|g}g~gahbhYbZbchYedhehfhgh0b1bhhihZe0ejhkhlhmhnh2b3boh1ephqhrhsh4b5bthuh2e3evhwhxhyhzh6b7bAh4eBhChDhEh8b9bFhGhHh5eIhJhKhLhMh6eNhOhPhQhRhSh7eThUhvdKmLmMmNm3dOm4dVhWhXh$dwd%df g h i a b c d 'd(dMdYhZh0h8e)d1h2h*d9e3h+d!e,dNd-de j k l m n 4h5h6h7ho 8hr 9h7d!h#exd$e#h$h%h'h(h!b#b)h%e*hkk$b%b+h'eOd(e,h-h.hAl/h'b(b:h)e;hlk)b*b{d*e+e=h,eZ 0 -e.e?h1 2 @h/ePd:e[h]h^hBl_h+b,b`h;e{hmk-b.b|d}d~ds t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 |h=eQd?e}h~haiClbi/b:bci@edink;b=bei[eRd]efigihiDlii?b@bji^ekiok[b]b_e.daebe`ecew x de{e|ey z li}eSd~eminioiElpi^b_bqiafripk`b{bbfcf7 8 9 ! df# $ % ' eeefffgf( ) hfsi* + iffeCdgejfkflfmf, - nfheofti. / : ; = ? @ [ ] ^ _ uipfTdqfviwixiFlyi|b}bzirfAiqk~bacsftfufvfwf` { iexf| } BiyfUdzfCiDiEiGlFibcccGiAfHirkdcec/djekeleA B C D meBfCfDf~ abEfbbcbIiFfVdGfJiKiLiHlMifcgcNiHfOiskhcicPiIfJfQiRiSiTijckcUiKfVilcmcWiLfWdMfXiYiZiIl0incoc1iNf2itkpcqcdb3iOfXdPf4i5i6iJl7ircsc8iQf9iuktcucneRfSf!iTfebfbUfVf#igbhb$iWfDdXf%i'i(iql)ivcwc*iYf+ivkxcycoepeqeE F G H gdzcAchdidjdkdI J ldBcmdudK L ,iZfEd0f-i.i/irl:iCcDc;i1f=iwkEcFc?i2fFd3f@i[i]isl^iGcHc_i4f`ixkIcJc5f:drese6fteM N ue7f8fO P {i9fGd!f|i}i~itlajKcLcbj#fcjykMcNc$f%fibjbkblb'fmbnbobpbve(f)f*fqbrb+fdjsbtbwe;dzd=dxeyezeAeQ R Be?dCe,fS T ubvbwbxbybzbAbBbCbej-fHd.ffjgjhjulijOcPcjj/fkjzkQcRc:f;f=f?f@fDbEbDe[fFbGblj]fId^fmjnjojvlpjScTcqj_frjAkUcVc@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsj~fJdagtjujvjwlwjWcXcxjbgyjBkYcZczjcgdgAjBjCjDj0c1cEjegFj2c3cGjfgKdggHjIjJjxlKj4c5cLjhgMjCk6c7cLbNjigYdjgOjPjQjKlRj8c9cSjkgTjDk!c#cUjlgLdmgVjWjXjylYj$c%cZjng0jEk'c(c1jogZdpg2j3j4jLl5j)c*c6jqg7jFk+c,c8jrg0dsg9j!j#jMl$j-c.c%jtg'jGk/c:c(jugAdvg)j*j+jHk,j;c=c-jwg.jIk?c@c/jxgBdyg:j;j=jJk?j[c]c@jzg[jKk^c_c]jAg1dBg^j_j`jPm{j`c{c|jCg}jLk|c}c~jDg2dEgakbkckQmdk~cadekFgfkMkbdcd
87 @staticmethod
88 cdef GraphDefinition _from_handle(GraphHandle h_graph):
89 """Create a GraphDefinition from an existing GraphHandle (internal use)."""
90 cdef GraphDefinition g = GraphDefinition.__new__(GraphDefinition) 2q IeMbNbObPb8dQbRbSbTbNlOlPlQlRlSlUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bf g h i a b c d MdNde !b#b$b%b'b(b)b*bZ 0 1 2 +b,b-b.bs t u v 3 4 5 6 /b:b;b=b?b@b[b]bw x y z ^b_b`b{b7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ |b}b~bac` { | } bcccdcecA B C D ~ abbbcbfcgchcicjckclcmcncocpcqcdbrcsctcucebfbgbhbvcwcxcycE F G H I J K L CcDcEcFcGcHcIcJcM N O P KcLcMcNcibjbkblbmbnbobpbqbrbsbtbQ R S T ubvbwbxbybzbAbBbCbOcPcQcRcDbEbFbGbScTcUcVcU V W X HbIbJbKbWcXcYcZc0c1c2c3c4c5c6c7cLb8c9c!c#c$c%c'c(c)c*c+c,c-c.c/c:c;c=c?c@c[c]c^c_c`c{c|c}c~cadbdcd
91 g._h_graph = h_graph 2q IeMbNbObPb8dQbRbSbTbNlOlPlQlRlSlUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bf g h i a b c d MdNde !b#b$b%b'b(b)b*bZ 0 1 2 +b,b-b.bs t u v 3 4 5 6 /b:b;b=b?b@b[b]bw x y z ^b_b`b{b7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ |b}b~bac` { | } bcccdcecA B C D ~ abbbcbfcgchcicjckclcmcncocpcqcdbrcsctcucebfbgbhbvcwcxcycE F G H I J K L CcDcEcFcGcHcIcJcM N O P KcLcMcNcibjbkblbmbnbobpbqbrbsbtbQ R S T ubvbwbxbybzbAbBbCbOcPcQcRcDbEbFbGbScTcUcVcU V W X HbIbJbKbWcXcYcZc0c1c2c3c4c5c6c7cLb8c9c!c#c$c%c'c(c)c*c+c,c-c.c/c:c;c=c?c@c[c]c^c_c`c{c|c}c~cadbdcd
92 return g 2q IeMbNbObPb8dQbRbSbTbNlOlPlQlRlSlUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bf g h i a b c d MdNde !b#b$b%b'b(b)b*bZ 0 1 2 +b,b-b.bs t u v 3 4 5 6 /b:b;b=b?b@b[b]bw x y z ^b_b`b{b7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ ] ^ _ |b}b~bac` { | } bcccdcecA B C D ~ abbbcbfcgchcicjckclcmcncocpcqcdbrcsctcucebfbgbhbvcwcxcycE F G H I J K L CcDcEcFcGcHcIcJcM N O P KcLcMcNcibjbkblbmbnbobpbqbrbsbtbQ R S T ubvbwbxbybzbAbBbCbOcPcQcRcDbEbFbGbScTcUcVcU V W X HbIbJbKbWcXcYcZc0c1c2c3c4c5c6c7cLb8c9c!c#c$c%c'c(c)c*c+c,c-c.c/c:c;c=c?c@c[c]c^c_c`c{c|c}c~cadbdcd
94 def __repr__(self) -> str:
95 return f"<GraphDefinition handle=0x{as_intptr(self._h_graph):x}>" 2Kl
97 def __eq__(self, other) -> bool:
98 if not isinstance(other, GraphDefinition): 28d9d!d#dNlOlPlQlRlSl'hqlrlsltlwe;dzd=dxeyezeAeQ R Be?dCe,fS T ulvlwlxlylHkJk
99 return NotImplemented 2'hqlrlsltlwe;dzd=dxeyezeAeQ R Be?dCe,fS T ulvlwlxl
100 return as_intptr(self._h_graph) == as_intptr((<GraphDefinition>other)._h_graph) 28d9d!d#dNlOlPlQlRlSl'hylHkJk
102 def __hash__(self) -> int:
103 return hash(as_intptr(self._h_graph)) 2AlBlClDlEliffeCdgejfkflfmf, - nfheofti. / FlGlHlIlJlLlMlHkJk
105 @property
106 def _entry(self) -> "GraphNode":
107 """Return the internal entry-point GraphNode (no dependencies)."""
108 cdef GraphNode n = GraphNode.__new__(GraphNode) 2RmSmTmUmVmq [dGg5dHgIgJgKeKgLgWm0mNkXmY OkPkQkRk6dSkTkUkVkWkXkYkZkgkhkikjk0k1k2k3k4k5k6kTlUlVlYmZmMgNgLeMeOgPgQgRgSgMbNbTgNeUgVgWgXgObPbYgZgOePe0g1g2g3g4gQbRb5gQe6g7g8g9gSbTb!g#g$gRe%g'g(g)g*gSe+g,g-g.gTeUe/g:g;g=g?gUbVb@gVe[g]g^g_gWbXb`g{gWeXe|g}g~gahbhYbZbchYedhehfhgh0b1bhhihZe0ejhkhlhmhnh2b3boh1ephqhrhsh4b5bthuh2e3evhwhxhyhzh6b7bAh4eBhChDhEh8b9bFhGhHh5eIhJhKhLhMh6eNhOhPhQhRhSh7eThUhvd4dVhWhXh$dwd%df g h i a b c d 'd(dMdYhZh0h8e)d1h2h*d9e3h+d!e,dNd-de j k l m n 4h5h6h7ho 8hr 9h7d!h#e$e#h$h%h(h!b#b)h%e*hkk$b%b+h'e(e,h-h.h/h'b(b:h)e;hlk)b*b{d*e+e=h,eZ 0 -e.e?h1 2 @h/e:e[h]h^h_h+b,b`h;e{hmk-b.b|d}d~ds t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 |h=e?e}h~haibi/b:bci@edink;b=bei[e]efigihiii?b@bji^ekiok[b]b_e.daebe`ecew x de{e|ey z li}e~eminioipi^b_bqiafripk`b{bbfcf7 8 9 ! df# $ % ' eeefffgf( ) hfsi* + iffegejfkflfmf, - nfheofti. / : ; = ? @ [ ] ^ _ uipfqfviwixiyi|b}bzirfAiqk~bacsftfufvfwf` { iexf| } BiyfzfCiDiEiFibcccGiAfHirkdcec/djekeleA B C D meBfCfDf~ abEfbbcbIiFfGfJiKiLiMifcgcNiHfOiskhcicPiIfJfQiRiSiTijckcUiKfVilcmcWiLfMfXiYiZi0incoc1iNf2itkpcqcdb3iOfPf4i5i6i7ircsc8iQf9iuktcucneRfSf!iTfebfbUfVf#igbhb$iWfXf%i'i(i)ivcwc*iYf+ivkxcycoepeqeE F G H gdzcAchdidjdkdI J ldBcmdudK L ,iZf0f-i.i/i:iCcDc;i1f=iwkEcFc?i2f3f@i[i]i^iGcHc_i4f`ixkIcJc5f:drese6fteM N ue7f8fO P {i9f!f|i}i~iajKcLcbj#fcjykMcNc$f%fibjbkblb'fmbnbobpbve(f)f*fqbrb+fdjsbtbwe;d=dxeyezeAeQ R Be?dCe,fS T ubvbwbxbybzbAbBbCbej-f.ffjgjhjijOcPcjj/fkjzkQcRc:f;f=f?f@fDbEbDe[fFbGblj]f^fmjnjojpjScTcqj_frjAkUcVc@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsj~fagtjujvjwjWcXcxjbgyjBkYcZczjcgdgAjBjCjDj0c1cEjegFj2c3cGjfgggHjIjJjKj4c5cLjhgMjCk6c7cLbNjigjgOjPjQjRj8c9cSjkgTjDk!c#cUjlgmgVjWjXjYj$c%cZjng0jEk'c(c1jogpg2j3j4j5j)c*c6jqg7jFk+c,c8jrgsg9j!j#j$j-c.c%jtg'jGk/c:c(jugvg)j*j+j,j;c=c-jwg.jIk?c@c/jxgyg:j;j=j?j[c]c@jzg[jKk^c_c]jAgBg^j_j`j{j`c{c|jCg}jLk|c}c~jDgEgakbkckdk~cadekFgfkMkbdcd
109 n._h_node = create_graph_node_handle(<cydriver.CUgraphNode>NULL, self._h_graph) 2RmSmTmUmVmq [dGg5dHgIgJgKeKgLgWm0mNkXmY OkPkQkRk6dSkTkUkVkWkXkYkZkgkhkikjk0k1k2k3k4k5k6kTlUlVlYmZmMgNgLeMeOgPgQgRgSgMbNbTgNeUgVgWgXgObPbYgZgOePe0g1g2g3g4gQbRb5gQe6g7g8g9gSbTb!g#g$gRe%g'g(g)g*gSe+g,g-g.gTeUe/g:g;g=g?gUbVb@gVe[g]g^g_gWbXb`g{gWeXe|g}g~gahbhYbZbchYedhehfhgh0b1bhhihZe0ejhkhlhmhnh2b3boh1ephqhrhsh4b5bthuh2e3evhwhxhyhzh6b7bAh4eBhChDhEh8b9bFhGhHh5eIhJhKhLhMh6eNhOhPhQhRhSh7eThUhvd4dVhWhXh$dwd%df g h i a b c d 'd(dMdYhZh0h8e)d1h2h*d9e3h+d!e,dNd-de j k l m n 4h5h6h7ho 8hr 9h7d!h#e$e#h$h%h(h!b#b)h%e*hkk$b%b+h'e(e,h-h.h/h'b(b:h)e;hlk)b*b{d*e+e=h,eZ 0 -e.e?h1 2 @h/e:e[h]h^h_h+b,b`h;e{hmk-b.b|d}d~ds t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 |h=e?e}h~haibi/b:bci@edink;b=bei[e]efigihiii?b@bji^ekiok[b]b_e.daebe`ecew x de{e|ey z li}e~eminioipi^b_bqiafripk`b{bbfcf7 8 9 ! df# $ % ' eeefffgf( ) hfsi* + iffegejfkflfmf, - nfheofti. / : ; = ? @ [ ] ^ _ uipfqfviwixiyi|b}bzirfAiqk~bacsftfufvfwf` { iexf| } BiyfzfCiDiEiFibcccGiAfHirkdcec/djekeleA B C D meBfCfDf~ abEfbbcbIiFfGfJiKiLiMifcgcNiHfOiskhcicPiIfJfQiRiSiTijckcUiKfVilcmcWiLfMfXiYiZi0incoc1iNf2itkpcqcdb3iOfPf4i5i6i7ircsc8iQf9iuktcucneRfSf!iTfebfbUfVf#igbhb$iWfXf%i'i(i)ivcwc*iYf+ivkxcycoepeqeE F G H gdzcAchdidjdkdI J ldBcmdudK L ,iZf0f-i.i/i:iCcDc;i1f=iwkEcFc?i2f3f@i[i]i^iGcHc_i4f`ixkIcJc5f:drese6fteM N ue7f8fO P {i9f!f|i}i~iajKcLcbj#fcjykMcNc$f%fibjbkblb'fmbnbobpbve(f)f*fqbrb+fdjsbtbwe;d=dxeyezeAeQ R Be?dCe,fS T ubvbwbxbybzbAbBbCbej-f.ffjgjhjijOcPcjj/fkjzkQcRc:f;f=f?f@fDbEbDe[fFbGblj]f^fmjnjojpjScTcqj_frjAkUcVc@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsj~fagtjujvjwjWcXcxjbgyjBkYcZczjcgdgAjBjCjDj0c1cEjegFj2c3cGjfgggHjIjJjKj4c5cLjhgMjCk6c7cLbNjigjgOjPjQjRj8c9cSjkgTjDk!c#cUjlgmgVjWjXjYj$c%cZjng0jEk'c(c1jogpg2j3j4j5j)c*c6jqg7jFk+c,c8jrgsg9j!j#j$j-c.c%jtg'jGk/c:c(jugvg)j*j+j,j;c=c-jwg.jIk?c@c/jxgyg:j;j=j?j[c]c@jzg[jKk^c_c]jAgBg^j_j`j{j`c{c|jCg}jLk|c}c~jDgEgakbkckdk~cadekFgfkMkbdcd
110 return n 2RmSmTmUmVmq [dGg5dHgIgJgKeKgLgWm0mNkXmY OkPkQkRk6dSkTkUkVkWkXkYkZkgkhkikjk0k1k2k3k4k5k6kTlUlVlYmZmMgNgLeMeOgPgQgRgSgMbNbTgNeUgVgWgXgObPbYgZgOePe0g1g2g3g4gQbRb5gQe6g7g8g9gSbTb!g#g$gRe%g'g(g)g*gSe+g,g-g.gTeUe/g:g;g=g?gUbVb@gVe[g]g^g_gWbXb`g{gWeXe|g}g~gahbhYbZbchYedhehfhgh0b1bhhihZe0ejhkhlhmhnh2b3boh1ephqhrhsh4b5bthuh2e3evhwhxhyhzh6b7bAh4eBhChDhEh8b9bFhGhHh5eIhJhKhLhMh6eNhOhPhQhRhSh7eThUhvd4dVhWhXh$dwd%df g h i a b c d 'd(dMdYhZh0h8e)d1h2h*d9e3h+d!e,dNd-de j k l m n 4h5h6h7ho 8hr 9h7d!h#e$e#h$h%h(h!b#b)h%e*hkk$b%b+h'e(e,h-h.h/h'b(b:h)e;hlk)b*b{d*e+e=h,eZ 0 -e.e?h1 2 @h/e:e[h]h^h_h+b,b`h;e{hmk-b.b|d}d~ds t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 |h=e?e}h~haibi/b:bci@edink;b=bei[e]efigihiii?b@bji^ekiok[b]b_e.daebe`ecew x de{e|ey z li}e~eminioipi^b_bqiafripk`b{bbfcf7 8 9 ! df# $ % ' eeefffgf( ) hfsi* + iffegejfkflfmf, - nfheofti. / : ; = ? @ [ ] ^ _ uipfqfviwixiyi|b}bzirfAiqk~bacsftfufvfwf` { iexf| } BiyfzfCiDiEiFibcccGiAfHirkdcec/djekeleA B C D meBfCfDf~ abEfbbcbIiFfGfJiKiLiMifcgcNiHfOiskhcicPiIfJfQiRiSiTijckcUiKfVilcmcWiLfMfXiYiZi0incoc1iNf2itkpcqcdb3iOfPf4i5i6i7ircsc8iQf9iuktcucneRfSf!iTfebfbUfVf#igbhb$iWfXf%i'i(i)ivcwc*iYf+ivkxcycoepeqeE F G H gdzcAchdidjdkdI J ldBcmdudK L ,iZf0f-i.i/i:iCcDc;i1f=iwkEcFc?i2f3f@i[i]i^iGcHc_i4f`ixkIcJc5f:drese6fteM N ue7f8fO P {i9f!f|i}i~iajKcLcbj#fcjykMcNc$f%fibjbkblb'fmbnbobpbve(f)f*fqbrb+fdjsbtbwe;d=dxeyezeAeQ R Be?dCe,fS T ubvbwbxbybzbAbBbCbej-f.ffjgjhjijOcPcjj/fkjzkQcRc:f;f=f?f@fDbEbDe[fFbGblj]f^fmjnjojpjScTcqj_frjAkUcVc@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsj~fagtjujvjwjWcXcxjbgyjBkYcZczjcgdgAjBjCjDj0c1cEjegFj2c3cGjfgggHjIjJjKj4c5cLjhgMjCk6c7cLbNjigjgOjPjQjRj8c9cSjkgTjDk!c#cUjlgmgVjWjXjYj$c%cZjng0jEk'c(c1jogpg2j3j4j5j)c*c6jqg7jFk+c,c8jrgsg9j!j#j$j-c.c%jtg'jGk/c:c(jugvg)j*j+j,j;c=c-jwg.jIk?c@c/jxgyg:j;j=j?j[c]c@jzg[jKk^c_c]jAgBg^j_j`j{j`c{c|jCg}jLk|c}c~jDgEgakbkckdk~cadekFgfkMkbdcd
112 def allocate(self, size_t size, *, device: "Device" | int | None = None,
113 memory_type: GraphMemoryType = GraphMemoryType.DEVICE,
114 peer_access: list["Device" | int] | None = None) -> "AllocNode":
115 """Add an entry-point memory allocation node (no dependencies).
117 See :meth:`GraphNode.allocate` for full documentation.
118 """
119 return self._entry.allocate(size, device=device, memory_type=memory_type, peer_access=peer_access) 2RmSmTmUmVm[dGg5dHgIgJgKeKgLgWmOkPkQkRkgkhkikjk0k1k2k3k4k5k6kTlUlVlMgNgMeOgPgQgTgNeUgVgWgXgYgZgPe0g1g2g5gQe6g7g8g9g!g#g$gRe%g'g(g)g*gSe+g,g-g.gUe/g:g;g@gVe[g]g^g_g`g{gXe|g}g~gchYedhehfhghhhih0ejhkhlhoh1ephqhrhshthuh3evhwhxhAh4eBhChDhEhFhGhHh5eIhJhKhLhMh6eNhOhQhRhSh7eThUhWhXh%dZh1h!h$e%h%e*h+h(e.h)e;h{d*e+e=h,eZ 0 -e.e?h1 2 @h:e^h;e{hndedqdfdtd|h?eai@ediei]ehi^eki_e.daebe`ecew x de{e|ey z li~eoiafrieeefffgf( ) hfsi* + ifgelfheofuiqfxirfAivfiexfBizfEiAfHi/djekeleA B C D meBfCfDf~ abEfbbcbIiGfLiHfOiPiJfSiKfViWiMfZiNf2i3iPf6iQf9ineRfSf!iTfebfbUfVf#igbhb$iXf(iYf+igdAcjdBcmd,i0f/i1f=i?i3f]i4f`i5f:drese6fteM N ue7f8fO P {i!f~i#fcjve(f)f*fqbrb+fdjsbtbwe=dze?dCeej.fhj/fkj?fDe[flj^foj_frj@dEeFeGeU V W X He`f{f|fHbIb}fJbKbsjagvjbgyjzjdgCjegFjGjggJjhgMjNjjgQjkgTjUjmgXjng0j1jpg4jqg7j8jsg#jtg'j(jvg+jwg.j/jyg=jzg[j]jBg`jCg}j~jEgckFgfk
121 def deallocate(self, dptr) -> "FreeNode":
122 """Add an entry-point memory free node (no dependencies).
124 See :meth:`GraphNode.deallocate` for full documentation.
125 """
126 return self._entry.deallocate(dptr) 2Vh
128 def memset(self, dst, value, size_t width, size_t height=1, size_t pitch=0) -> "MemsetNode":
129 """Add an entry-point memset node (no dependencies).
131 See :meth:`GraphNode.memset` for full documentation.
132 """
133 return self._entry.memset(dst, value, width, height, pitch) 1por
135 def launch(self, config, kernel, *args) -> "KernelNode":
136 """Add an entry-point kernel launch node (no dependencies).
138 See :meth:`GraphNode.launch` for full documentation.
139 """
140 return self._entry.launch(config, kernel, *args) 26dVkWkXkYkZkYmZmLeOeTeWeZe2e$da b c d Md*d9e3h+d!e,dNdj k l m n o 7d#e)h'e:h{d-e/e`h|d}d~ds t u v ddsd=eci[eji.dde}eqieefenfpfzisftfufvfwf` { iexf| } yfGi/dmeFfNiIfUiLf1iOf8ineUfWf*ioepeqeE F G H zcldZf;i2f_i:due9fbjve;dBe-fjj:f;f=f?f@fDbEbDe[fFbGb]fqj@dHe~fxjcgEjfgLjigSjlgZjog6jrg%jug-jxg@jAg|jDgek
142 def empty(self) -> "EmptyNode":
143 """Add an entry-point empty node (no dependencies).
145 Returns
146 -------
147 EmptyNode
148 A new EmptyNode with no dependencies.
149 """
150 return self._entry.join() 2Y Ph4d4h5h6h7h8h9h
152 def join(self, *nodes) -> "EmptyNode":
153 """Create an empty node that depends on all given nodes.
155 Parameters
156 ----------
157 *nodes : GraphNode
158 Nodes to merge.
160 Returns
161 -------
162 EmptyNode
163 A new EmptyNode that depends on all input nodes.
164 """
165 return self._entry.join(*nodes) 2KegkhkikjkTlUlVlMeNePeQeReSeUeVeXeYe0e1e3e4e5e6e7ej k l m n $e%e(e)e.e:e;eedfd?e@e]e^e_e.daebe`ecew x de{e|ey z ~eafhfgeheqfrfiezfAf/djekeleA B C D EfGfHfJfKfMfNfPfQfVfXfYfAcBc0f1f3f4f5f:drese6fteM N ue7f8fO P !f#f+f=d?d.f/fDe^f_f@dEeFeGeU V W X }fagbgdgeggghgjgkgmgngpgqgsgtgvgwgygzgBgCgEgFg
167 def memcpy(self, dst, src, size_t size) -> "MemcpyNode":
168 """Add an entry-point memcpy node (no dependencies).
170 See :meth:`GraphNode.memcpy` for full documentation.
171 """
172 return self._entry.memcpy(dst, src, size)
174 def embed(self, child: GraphDefinition) -> "ChildGraphNode":
175 """Add an entry-point child graph node (no dependencies).
177 See :meth:`GraphNode.embed` for full documentation.
178 """
179 return self._entry.embed(child) 26dLeOeTeWeZe2eMdNd#e'e{d/e|d}d~ds t u v dd=e[e.d}eeefepfsfyf/dmeFfIfLfOfneWfoepeqeE F G H zcZf2f:d9fve;d-f:f]f@dHe~fcgfgiglgogrgugxgAgDg
181 def record(self, event) -> "EventRecordNode":
182 """Add an entry-point event record node (no dependencies).
184 See :meth:`GraphNode.record` for full documentation.
185 """
186 return self._entry.record(event) 2SkYh0h8e)d#h,h*e[h|dod}hfiaemibfcf7 8 9 ! efjfvitfCijeBfJiQiXi4iRf%ioehd-i@ire|i$f%fibjbkblb(fxefj;fmjEe`ftjAjHjOjVj2j9j)j:j^jak
188 def wait(self, event) -> "EventWaitNode":
189 """Add an entry-point event wait node (no dependencies).
191 See :meth:`GraphNode.wait` for full documentation.
192 """
193 return self._entry.wait(event) 22h$h-h+e]h}dpd~hgibenibfdf# $ % ' ffkfwiufDikeCfKiRiYi5iSf'ipeid.i[ise}i$f'fmbnbobpb)fyegj=fnjFe{fujBjIjPjWj3j!j*j;j_jbk
195 def callback(self, fn, *, user_data=None) -> "HostCallbackNode":
196 """Add an entry-point host callback node (no dependencies).
198 See :meth:`GraphNode.callback` for full documentation.
199 """
200 return self._entry.callback(fn, user_data=user_data) 2NkXmTkUkRgSg3g4g=g?gahbhmhnhyhzh'd(d-d(h/h,e_h~drdbiiicepicfdfgfmf: ; = ? yiwfFileDfMiTi0i7iTf)iqekd:i^iteaj%f'f*fAeubvbwbxbij@fpjGe|fwjDjKjRjYj5j$j,j?j{jdk
202 def create_condition(self, default_value: int | None = None) -> GraphCondition:
203 """Create a condition variable for use with conditional nodes.
205 The returned :class:`GraphCondition` object is passed to conditional-node
206 builder methods. Its value is controlled at runtime by device code
207 via ``cudaGraphSetConditional``.
209 Parameters
210 ----------
211 default_value : int, optional
212 The default value to assign to the condition.
213 If None, no default is assigned.
215 Returns
216 -------
217 GraphCondition
218 A condition variable for controlling conditional execution.
219 """
220 cdef cydriver.CUgraphConditionalHandle c_handle
221 cdef unsigned int flags = 0 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
222 cdef unsigned int default_val = 0 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
224 if default_value is not None: 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
225 default_val = <unsigned int>default_value 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvdwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
226 flags = cydriver.CU_GRAPH_COND_ASSIGN_DEFAULT 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvdwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
228 cdef cydriver.CUcontext ctx = NULL 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
229 with nogil: 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
230 HANDLE_RETURN(cydriver.cuCtxGetCurrent(&ctx)) 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
231 HANDLE_RETURN(cydriver.cuGraphConditionalHandleCreate( 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
232 &c_handle, as_cu(self._h_graph), ctx, default_val, flags))
234 return GraphCondition._from_handle(c_handle) 2q MbNbObPbQbRbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9bvd3dwdf g h i a b c d e xd!b#b$b%bOd'b(b)b*bZ 0 1 2 Pd+b,b-b.bs t u v ndddedodpdqdrd3 4 sdfdtdyd5 6 Qd/b:b;b=bRd?b@b[b]bw x y z Sd^b_b`b{b7 8 9 ! # $ % ' ( ) * + Cd, - . / : ; = ? @ [ ] ^ _ Td|b}b~bac` { | } UdbcccdcecA B C D ~ abbbcbVdfcgchcicjckclcmcWdncocpcqcdbXdrcsctcucebfbgbhbDdvcwcxcycE F G H gdzcAchdidjdkdI J ldBcmdudK L EdCcDcEcFcFdGcHcIcJcM N O P GdKcLcMcNcibjbkblbmbnbobpbqbrbsbtbzdQ R S T ubvbwbxbybzbAbBbCbHdOcPcQcRcDbEbFbGbIdScTcUcVcU V W X HbIbJbKbJdWcXcYcZc0c1c2c3cKd4c5c6c7cLbYd8c9c!c#cLd$c%c'c(cZd)c*c+c,c0d-c.c/c:cAd;c=c?c@cBd[c]c^c_c1d`c{c|c}c2d~cadbdcd
236 def if_then(self, condition: GraphCondition) -> "IfNode":
237 """Add an entry-point if-conditional node (no dependencies).
239 See :meth:`GraphNode.if_then` for full documentation.
240 """
241 return self._entry.if_then(condition) 2q NbRbVbZb3b7bvdf a #b(b0 ,bt 4 :b@bx _b8 $ ) - ; ] ^ _ }b{ ccB abgckcocscfbwcF J DcHcN LcjbnbrbR vbAbBbCbPcEbTcV IbXc1c5c9c%c*c.c=c]c{cad
243 def if_else(self, condition: GraphCondition) -> "IfElseNode":
244 """Add an entry-point if-else conditional node (no dependencies).
246 See :meth:`GraphNode.if_else` for full documentation.
247 """
248 return self._entry.if_else(condition) 2MbQbUbYb2b6bg b !b'bZ +bs 3 /b?bw ^b7 # ( , : @ [ ] |b` bcA ~ fcjcncrcebvcE I CcGcM KcibmbqbQ ubybzbAbOcDbScU HbWc0c4c8c$c)c-c;c[c`c~c
250 def while_loop(self, condition: GraphCondition) -> "WhileNode":
251 """Add an entry-point while-loop conditional node (no dependencies).
253 See :meth:`GraphNode.while_loop` for full documentation.
254 """
255 return self._entry.while_loop(condition) 2PbTbXb1b5b9bi d e %b*b2 .bv 6 =b]bz {b! ' + / ? [ _ ac} ecD cbicmcqcdbuchbycH L FcJcP NclbpbtbT xbzbCbRcGbVcX KbZc3c7cLb#c(c,c:c@c_c}ccd
257 def switch(self, condition: GraphCondition, unsigned int count) -> "SwitchNode":
258 """Add an entry-point switch conditional node (no dependencies).
260 See :meth:`GraphNode.switch` for full documentation.
261 """
262 return self._entry.switch(condition, count) 2ObSbWb0b4b8bwdh c $b)b1 -bu 5 ;b[by `b9 % * . = @ ^ ~b| dcC bbhclcpcdbtcgbxcG K EcIcO MckbobsbS wbybBbQcFbUcW JbYc2c6cLb!c'c+c/c?c^c|cbd
264 def instantiate(self, options=None):
265 """Instantiate the graph definition into an executable Graph.
267 Parameters
268 ----------
269 options : :obj:`~graph.GraphCompleteOptions`, optional
270 Customizable dataclass for graph instantiation options.
272 Returns
273 -------
274 Graph
275 An executable graph that can be launched on a stream.
276 """
277 from cuda.core.graph._graph_builder import _instantiate_graph 2NkOkPkQkRk6dSkTkUkVkWkXkYkZkgkhkikjk0k1k2k3k1m2m3m4k5k6k$d%d'd(d)d*d+d,d-dj k l m n o 7d
279 return _instantiate_graph( 2NkOkPkQkRk6dSkTkUkVkWkXkYkZkgkhkikjk0k1k2k3k1m2m3m4k5k6k$d%d'd(d)d*d+d,d-dj k l m n o 7d
280 driver.CUgraph(as_intptr(self._h_graph)), options) 2NkOkPkQkRk6dSkTkUkVkWkXkYkZkgkhkikjk0k1k2k3k1m2m3m4k5k6k$d%d'd(d)d*d+d,d-dj k l m n o 7d
282 def debug_dot_print(self, path: str, options=None) -> None:
283 """Write a GraphViz DOT representation of the graph to a file.
285 Parameters
286 ----------
287 path : str
288 File path for the DOT output.
289 options : GraphDebugPrintOptions, optional
290 Customizable options for the debug print.
291 """
292 from cuda.core.graph._graph_builder import GraphDebugPrintOptions 2[dGg5d
294 cdef unsigned int flags = 0 2[dGg5d
295 if options is not None: 2[dGg5d
296 if not isinstance(options, GraphDebugPrintOptions): 2Gg5d
297 raise TypeError("options must be a GraphDebugPrintOptions instance") 2Gg
298 flags = options._to_flags() 25d
300 cdef bytes path_bytes = path.encode('utf-8') 2[d5d
301 cdef const char* c_path = path_bytes 2[d5d
302 with nogil: 2[d5d
303 HANDLE_RETURN(cydriver.cuGraphDebugDotPrint(as_cu(self._h_graph), c_path, flags)) 2[d5d
305 def nodes(self) -> set:
306 """Return all nodes in the graph.
308 Returns
309 -------
310 set of GraphNode
311 All nodes in the graph.
312 """
313 cdef vector[cydriver.CUgraphNode] nodes_vec
314 nodes_vec.resize(128) 2q Y Ie7k8k8d9k!k#k$k9d!d%k'k(k)k*k+kJe#d,k-k.kzl/k:k;k=k?k@k[k]k^k_k`k{k|k}k~kalblcldlelflglhliljlklllml4df g h i a b c d MdNde j k l m n o r
315 cdef size_t num_nodes = 128 2q Y Ie7k8k8d9k!k#k$k9d!d%k'k(k)k*k+kJe#d,k-k.kzl/k:k;k=k?k@k[k]k^k_k`k{k|k}k~kalblcldlelflglhliljlklllml4df g h i a b c d MdNde j k l m n o r
317 with nogil: 2q Y Ie7k8k8d9k!k#k$k9d!d%k'k(k)k*k+kJe#d,k-k.kzl/k:k;k=k?k@k[k]k^k_k`k{k|k}k~kalblcldlelflglhliljlklllml4df g h i a b c d MdNde j k l m n o r
318 HANDLE_RETURN(cydriver.cuGraphGetNodes(as_cu(self._h_graph), nodes_vec.data(), &num_nodes)) 2q Y Ie7k8k8d9k!k#k$k9d!d%k'k(k)k*k+kJe#d,k-k.kzl/k:k;k=k?k@k[k]k^k_k`k{k|k}k~kalblcldlelflglhliljlklllml4df g h i a b c d MdNde j k l m n o r
320 if num_nodes == 0: 2q Y Ie7k8k8d9k!k#k$k9d!d%k'k(k)k*k+kJe#d,k-k.kzl/k:k;k=k?k@k[k]k^k_k`k{k|k}k~kalblcldlelflglhliljlklllml4df g h i a b c d MdNde j k l m n o r
321 return set() 2zlf g h i e
323 if num_nodes > 128: 2q Y Ie7k8k8d9k!k#k$k9d!d%k'k(k)k*k+kJe#d,k-k.k/k:k;k=k?k@k[k]k^k_k`k{k|k}k~kalblcldlelflglhliljlklllml4da b c d MdNde j k l m n o r
324 nodes_vec.resize(num_nodes)
325 with nogil:
326 HANDLE_RETURN(cydriver.cuGraphGetNodes(as_cu(self._h_graph), nodes_vec.data(), &num_nodes))
328 return {GraphNode._create(self._h_graph, nodes_vec[i]) for i in range(num_nodes)} 2q Y Ie7k8k8d9k!k#k$k9d!d%k'k(k)k*k+kJe#d,k-k.k/k:k;k=k?k@k[k]k^k_k`k{k|k}k~kalblcldlelflglhliljlklllml4da b c d MdNde j k l m n o r
330 def edges(self) -> set:
331 """Return all edges in the graph as (from_node, to_node) pairs.
333 Returns
334 -------
335 set of tuple
336 Each element is a (from_node, to_node) pair representing
337 a dependency edge in the graph.
338 """
339 cdef vector[cydriver.CUgraphNode] from_nodes
340 cdef vector[cydriver.CUgraphNode] to_nodes
341 from_nodes.resize(128) 2]d^dnlol_d`dplY j k l m n o r
342 to_nodes.resize(128) 2]d^dnlol_d`dplY j k l m n o r
343 cdef size_t num_edges = 128 2]d^dnlol_d`dplY j k l m n o r
345 with nogil: 2]d^dnlol_d`dplY j k l m n o r
346 IF CUDA_CORE_BUILD_MAJOR >= 13:
347 HANDLE_RETURN(cydriver.cuGraphGetEdges( 2]d^dnlol_d`dplY j k l m n o r
348 as_cu(self._h_graph), from_nodes.data(), to_nodes.data(), NULL, &num_edges))
349 ELSE:
350 HANDLE_RETURN(cydriver.cuGraphGetEdges(
351 as_cu(self._h_graph), from_nodes.data(), to_nodes.data(), &num_edges))
353 if num_edges == 0: 2]d^dnlol_d`dplY j k l m n o r
354 return set() 2nlolplr
356 if num_edges > 128: 2]d^d_d`dY j k l m n o r
357 from_nodes.resize(num_edges)
358 to_nodes.resize(num_edges)
359 with nogil:
360 IF CUDA_CORE_BUILD_MAJOR >= 13:
361 HANDLE_RETURN(cydriver.cuGraphGetEdges(
362 as_cu(self._h_graph), from_nodes.data(), to_nodes.data(), NULL, &num_edges))
363 ELSE:
364 HANDLE_RETURN(cydriver.cuGraphGetEdges(
365 as_cu(self._h_graph), from_nodes.data(), to_nodes.data(), &num_edges))
367 return { 2]d^d_d`dY j k l m n o r
368 (GraphNode._create(self._h_graph, from_nodes[i]), 2]d^d_d`dY j k l m n o r
369 GraphNode._create(self._h_graph, to_nodes[i])) 2]d^d_d`dY j k l m n o r
370 for i in range(num_edges) 2]d^d_d`dY j k l m n o r
371 }
373 @property
374 def handle(self) -> driver.CUgraph:
375 """Return the underlying driver CUgraph handle."""
376 return as_py(self._h_graph) 28m8e9e!e7d'h