Skip to content

Instantly share code, notes, and snippets.

@soulitzer

soulitzer/gist:4da95730e40d814aa0c64cdff5c48571

Last active June 19, 2023 20:17
Show Gist options
  • Save soulitzer/4da95730e40d814aa0c64cdff5c48571 to your computer and use it in GitHub Desktop.
Save soulitzer/4da95730e40d814aa0c64cdff5c48571 to your computer and use it in GitHub Desktop.
Download ZIP
fx graph before inlining modules
Raw
gistfile1.txt
(pytorch1) jw3468-mbp:pytorch1 jw3468$ python test/dynamo/test_modules.py -k TestTemp -v
test_dynamo_inline_module_nn_AdaptiveAvgPool1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AdaptiveAvgPool2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AdaptiveAvgPool3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AdaptiveMaxPool1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AdaptiveMaxPool2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AdaptiveMaxPool3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AvgPool1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AvgPool2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_AvgPool3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_BatchNorm1d_eval_mode_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_BatchNorm1d_train_mode_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_BatchNorm2d_eval_mode_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_BatchNorm2d_train_mode_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_BatchNorm3d_eval_mode_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_BatchNorm3d_train_mode_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_Bilinear_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_CELU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ConstantPad1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ConstantPad2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ConstantPad3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Conv1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Conv2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Conv3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ConvTranspose1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ConvTranspose2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ConvTranspose3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_CrossEntropyLoss_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ELU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Embedding_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_FractionalMaxPool2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_FractionalMaxPool3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_GELU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_GLU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_GRUCell_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_GRU_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
unimplemented [('TorchDynamo purposely graph breaks on RNN, GRU, LSTMs', 1)]
expected failure
test_dynamo_inline_module_nn_GRU_train_mode_cpu_float32 (__main__.TestTempCPU) ...
unimplemented [('TorchDynamo purposely graph breaks on RNN, GRU, LSTMs', 1)]
expected failure
test_dynamo_inline_module_nn_GaussianNLLLoss_cpu_float32 (__main__.TestTempCPU) ...
unimplemented [('data dependent operator: aten._local_scalar_dense.default', 1)]
expected failure
test_dynamo_inline_module_nn_GroupNorm_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Hardshrink_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Hardswish_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Hardtanh_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_InstanceNorm1d_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_InstanceNorm1d_train_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_InstanceNorm2d_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_InstanceNorm2d_train_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_InstanceNorm3d_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_InstanceNorm3d_train_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_L1Loss_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LPPool1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LPPool2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LSTMCell_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
------------- ----------- --------------------------- --------------------------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_0_ L_args_1_0_ () {}
placeholder l_args_1_1_ L_args_1_1_ () {}
call_module m m (l_args_0_, (l_args_1_0_, l_args_1_1_)) {}
call_function getitem <built-in function getitem> (m, 0) {}
call_function getitem_1 <built-in function getitem> (m, 1) {}
output output output ((getitem, getitem_1),) {}
stats [('calls_captured', 3), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LSTM_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
unimplemented [('TorchDynamo purposely graph breaks on RNN, GRU, LSTMs', 1)]
expected failure
test_dynamo_inline_module_nn_LSTM_train_mode_cpu_float32 (__main__.TestTempCPU) ...
unimplemented [('TorchDynamo purposely graph breaks on RNN, GRU, LSTMs', 1)]
expected failure
test_dynamo_inline_module_nn_LayerNorm_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LazyConv1d_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_LazyConv2d_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_LazyConv3d_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_LazyConvTranspose1d_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_LazyConvTranspose2d_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_LazyConvTranspose3d_cpu_float32 (__main__.TestTempCPU) ... skipped 'skipping lazy module'
test_dynamo_inline_module_nn_LeakyReLU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Linear_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------------- --------------- ------- --------------------------
placeholder l_kwargs_input_ L_kwargs_input_ () {}
call_module m m () {'input': l_kwargs_input_}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LocalResponseNorm_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LogSigmoid_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_LogSoftmax_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_MaxPool1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_MaxPool2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_MaxPool3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Mish_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_MultiheadAttention_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
------------- --------- --------------------------- --------------------------------- ---------------------------------------------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
placeholder l_args_2_ L_args_2_ () {}
call_module m m (l_args_0_, l_args_1_, l_args_2_) {'key_padding_mask': None, 'attn_mask': None}
call_function getitem <built-in function getitem> (m, 0) {}
call_function getitem_1 <built-in function getitem> (m, 1) {}
output output output ((getitem, getitem_1),) {}
stats [('calls_captured', 3), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_MultiheadAttention_train_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
------------- --------- --------------------------- --------------------------------- ---------------------------------------------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
placeholder l_args_2_ L_args_2_ () {}
call_module m m (l_args_0_, l_args_1_, l_args_2_) {'key_padding_mask': None, 'attn_mask': None}
call_function getitem <built-in function getitem> (m, 0) {}
call_function getitem_1 <built-in function getitem> (m, 1) {}
output output output ((getitem, getitem_1),) {}
stats [('calls_captured', 3), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_NLLLoss_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_PReLU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_RNNCell_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_RNN_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
unimplemented [('TorchDynamo purposely graph breaks on RNN, GRU, LSTMs', 1)]
expected failure
test_dynamo_inline_module_nn_RNN_train_mode_cpu_float32 (__main__.TestTempCPU) ...
unimplemented [('TorchDynamo purposely graph breaks on RNN, GRU, LSTMs', 1)]
expected failure
test_dynamo_inline_module_nn_ReLU6_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ReLU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ReflectionPad1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ReflectionPad2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ReflectionPad3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ReplicationPad1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ReplicationPad2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ReplicationPad3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_SELU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_SiLU_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Sigmoid_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Softmax2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Softmax_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Softmin_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Softplus_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Softshrink_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Softsign_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Tanh_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Tanhshrink_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_Threshold_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_TransformerDecoderLayer_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- --------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_TransformerEncoderLayer_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_TransformerEncoderLayer_train_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_TransformerEncoder_eval_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
unexpected success
test_dynamo_inline_module_nn_TransformerEncoder_train_mode_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
unexpected success
test_dynamo_inline_module_nn_Transformer_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ---------------------- ------------------------------------------------------------------------------------------------
placeholder l_args_0_ L_args_0_ () {}
placeholder l_args_1_ L_args_1_ () {}
call_module m m (l_args_0_, l_args_1_) {'tgt_mask': None, 'src_mask': None, 'tgt_key_padding_mask': None, 'src_key_padding_mask': None}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
unexpected success
test_dynamo_inline_module_nn_ZeroPad1d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ZeroPad2d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
test_dynamo_inline_module_nn_ZeroPad3d_cpu_float32 (__main__.TestTempCPU) ...
opcode name target args kwargs
----------- --------- --------- ------------ --------
placeholder l_args_0_ L_args_0_ () {}
call_module m m (l_args_0_,) {}
output output output ((m,),) {}
stats [('calls_captured', 1), ('unique_graphs', 1)]
ok
----------------------------------------------------------------------
Ran 105 tests in 0.770s
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment