΅ib S SK Jr S SKJrJr S SKrS SKrS SKrS SKJ r S\
S\ \
4S jr S\R R 4S jrS
\
S\\\ \ 4 4S jrS rS
rSSS.S\4S jjr\S 5 rS\4S jrS\4S jrS\4S jrg) )contextmanager)AnycastN)Timerfilenamereturnc SnSnS n/ n[ U 5 nUR 5 R U5 n[ U5 H u pxUS:X a M
UR U5 n U S:X a M& US U n
XgS-
R 5 S nU
R SS9 V s/ s H o[
U5 S PM n n UR SR U 5 5 M S S S 5 U$ s sn f ! , ( d f U$ = f) Nzzr T)keepends ) openreadsplit enumeratefind
splitlineslenappendjoin)
r BEGINENDpfxgraphsf
split_strsi split_strend_locsxliness
U/home/james-whalen/.local/lib/python3.13/site-packages/torch/utils/jit/log_extract.py
extract_irr$ s E
C
C
F
h 1VVX^^E*
%j 1LAAv nnS)G"}(7#A U#..04C+,<<<+FG+Fas3xy\+FEG MM"''%. ) 2
M H
Ms A?C C*&C C
C.inp_typec U R 5 nU R 5 nU R 5 nU R 5 nUc [ S5 eUc [ S5 eUc [ S5 eUc [ S5 e[
R " XX4S9$ )NzFmake_tensor_from_type: 'size' is None (inp_type.sizes() returned None)zJmake_tensor_from_type: 'stride' is None (inp_type.strides() returned None)zImake_tensor_from_type: 'device' is None (inp_type.device() returned None)zGmake_tensor_from_type: 'dtype' is None (inp_type.dtype() returned None))sizestridedevicedtype)sizesstridesr) r* AssertionErrortorch
empty_strided)r% r' r( r) r* s r# make_tensor_from_typer0 s >> D
F
__
F NN E | eff
~ ijj
~ hii } fgg D TT irc [ R R U SS9nUR 5 / nUR 5 GH n[ UR
5 [ R R 5 ( a( UR [ R " SS5 5 Mc [ UR
5 [ R R 5 ( a( UR [ R " SS5 5 M [ UR
5 [ R R 5 ( aO [ [ R R UR
5 5 nUR [ U5 5 GMH [ UR
5 [ R R 5 ( a, UR [ R " SS5 S:H 5 GM [# SUR
5 35 e [ R R% SU5 n[ R R' UR( 5 XR4$ ) NT)parse_tensor_constantsg?d r r z,A default value is not implemented for type forward)r. _Cparse_irmakeMultiOutputIntoTupleinputs
isinstancetype FloatTyper randomuniformIntTyperandint
TensorTyper r0 BoolTypeNotImplementedError_create_function_from_graph!_jit_pass_erase_shape_informationgraph)r2 rG r: inp
tensorTypefuncs r# load_graph_and_inputsrK - s| HH b >E ""$
F||~ chhj%(("4"4 5 5 MM&..S1 2
EHH$4$4
5
5 MM&..C0 1
EHH$7$7
8
8 ehh 1 1388:>J MM/
; <
EHH$5$5
6
6 MM&..A.!3 4%(TUXU]U]U_T`&ab b 88 / / 5 AD HH..tzz: >r1 c X [ SXS.S9nUR 5 nUR S- $ )Nz fn(*inputs))fnr: )stmtglobals )r blocked_autorangemedian)rM r: test_runsttimess r# time_cudarV B s.
=*GHA
!E <<$ r1 c [ R " 5 n[ U5 H nU " U6 M
[ R " 5 nXS-
U- S- $ )NrP )time perf_counterrange)rM r: rS r _es r# time_cpur] G sF A
9
F A
EY %%r1
) warmup_runsrS c 6 [ U 5 u pE[ U5 H nU" U6 M
S nU H= n[ U[ R 5 ( d M$ UR
R S:H n O Uc [ S5 eU( a [ XAU5 nU$ [ XAU5 nU$ )NcpuzNo tensor found in inputs)
rK rZ r; r. Tensorr) r< r- r] rV ) r2 r: r` rS rG r[ is_cpuinputouts r# run_testrg N s $R(HE
;
v F eU\\ * *\\&&%/F ~ 89906(5)
,C J =FeU^<_C Jr1 c / # [ R R S5 n S v [ R R U5 g ! [ R R U5 f = f7f)NF)r. r7 _get_graph_executor_optimize)argskwargs old_optimizes r# no_fuserrm ^ sE 8888?L<
--l;--l;s A+A A+!A((A+c b [ 5 [ X5 sS S S 5 $ ! , ( d f g = f)N)rm rg r2 r: s r# run_baseline_no_fusionrp f s #
s
.c U( a S/OS/n[ R R U5 n[ R R S5 [ X5 sS S S 5 [ R R U5 $ ! , ( d f O= f [ R R U5 g ! [ R R W5 f = f)N)DYNAMICr^ )STATICr^ fuser1)r. jitset_fusion_strategyfuserrg )r2 r: dynamicstrat old_strats r# run_nncr{ k s 1%, !>2BII11%8
YY__X
&B' '
& %%i0 '
&
& %%i0 %%i0s$ A B3
B B3
B B3 3!Cc [ R R S5 [ X5 sS S S 5 $ ! , ( d f g = f)Nfuser2)r. ru rw rg ro s r# run_nvfuserr~ t s& "#
# " "s 5
A)
contextlibr typingr r r> r. rX torch.utils.benchmarkr strlistr$ r7 rB r0 tuplerK rV r] floatrg rm rp r{ r~ r1 r# r s %
' c *
UEHH$7$7
U c eCcN&; *
& )+b U < <$% $
1E 1$u $r1