[Th#lSSKrSSKrSSKJr SSKJr SSKJr SSKJ r SSK J r "SS\5r S rg) N)FakeTensorProp)CapabilityBasedPartitioner)OperatorSupport)CALLABLE_NODE_OPS)_pytreecN\rSrSrS\R R S\4SjrSr g)CudaGraphsSupport nodereturnc^UR[;agUR[RR R R4;agUR[R4;agSmSnU4SjnURH)n[R"XC"UR55 M+ [R"XC"UR55 T(+$)NFTc"SU;aUS$US$)Nval fake_result)metas [/var/www/auris/envauris/lib/python3.13/site-packages/torch/fx/passes/backends/cudagraphs.pymeta_fk4CudaGraphsSupport.is_node_supported..meta_fks"'4-4; HT-5H Hc>[U[R5(aURRS:waSmggg)NcudaT) isinstancetorchTensordevicetype)tfound_not_cudas r find_not_cuda:CudaGraphsSupport.is_node_supported..find_not_cudas1!U\\**qxx}}/F!%0G*r)oprtargetropsatenembedding_dense_backwarddefaultoperatorgetitemall_input_nodespytree tree_map_r)self submodulesr rr nrs @ris_node_supported#CudaGraphsSupport.is_node_supporteds 77+ + ;;599>>BBJJK K ;;8++, , I & %%A   ]GAFFO <&  (:; "!!rrN) __name__ __module__ __qualname____firstlineno__rfxNodeboolr0__static_attributes__rrrr r s"%((--"D"rr c[U5R"U6 [5n[XSS9nUR 5nUR U5nU$)z Partition an FX graph into sub-GraphModules that can be validly run under CUDA graphs. For a subgraph to be runnable under CUDA, all of the operations must involve CUDA tensors only/ T)allows_single_node_partition)r propagater rpropose_partitionsfuse_partitions)gminputs supported_ops partitioner partitions fused_graphs rpartition_cudagraphsrE-sV2  &)%'M- K//1J--j9K r)r(r torch.fx.passes.fake_tensor_propr!torch.fx.passes.infra.partitionerr torch.fx.passes.operator_supportrtorch.fx.passes.tools_commonr torch.utilsrr+r rErrrrKs, ;H<:)""Br