V1219 16:33:18.212000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:1234] skipping: _wrapped_call_impl (reason: in skipfiles, file: /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py) V1219 16:33:18.212000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:1234] skipping: _call_impl (reason: in skipfiles, file: /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] torchdynamo start compiling forward /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:183, stack (elided 4 frames): V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py", line 227, in V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] aten_dialect = export(simulator, example_inputs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/__init__.py", line 270, in export V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] return _export( V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 990, in wrapper V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] ep = fn(*args, **kwargs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/exported_program.py", line 114, in wrapper V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] return fn(*args, **kwargs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 1880, in _export V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] export_artifact = export_func( # type: ignore[operator] V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 1224, in _strict_export V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] return _strict_export_lower_to_aten_ir( V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 1252, in _strict_export_lower_to_aten_ir V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] gm_torch_level = _export_to_torch_ir( V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 560, in _export_to_torch_ir V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] gm_torch_level, _ = torch._dynamo.export( V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 1432, in inner V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] result_traced = opt_f(*args, **kwargs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] return self._call_impl(*args, **kwargs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] return forward_call(*args, **kwargs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 465, in _fn V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] return fn(*args, **kwargs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] return self._call_impl(*args, **kwargs) V1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py:864] [0/0] I1219 16:33:18.215000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:859] [0/0] ChromiumEventLogger initialized with id d4706065-cfc9-4744-8922-8c3f00f352a2 I1219 16:33:18.231000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/logging.py:57] [0/0] Step 1: torchdynamo start tracing forward /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:183 V1219 16:33:18.231000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:2498] [0/0] create_env /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/onnxscript/converter.py:820: FutureWarning: 'onnxscript.values.Op.param_schemas' is deprecated in version 0.1 and will be removed in the future. Please use '.op_signature' instead. param_schemas = callee.param_schemas() /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/onnxscript/converter.py:820: FutureWarning: 'onnxscript.values.OnnxFunction.param_schemas' is deprecated in version 0.1 and will be removed in the future. Please use '.op_signature' instead. param_schemas = callee.param_schemas() V1219 16:33:18.585000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/output_graph.py:2076] [0/0] create_graph_input L_particle_type_ L['particle_type'] V1219 16:33:18.586000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/builder.py:2702] [0/0] wrap_to_fake L['particle_type'] (858,) StatefulSymbolicContext(dynamic_sizes=[], dynamic_strides=[], constraint_sizes=[StrictMinMaxConstraint(warn_only=False, vr=VR[858, 858])], constraint_strides=[None], view_base_context=None, tensor_source=LocalSource(local_name='particle_type', cell_or_freevar=False), shape_env_to_source_to_symbol_cache={}) V1219 16:33:18.587000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/output_graph.py:2076] [0/0] create_graph_input L_position_sequence_ L['position_sequence'] V1219 16:33:18.587000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/builder.py:2702] [0/0] wrap_to_fake L['position_sequence'] (858, 6, 2) StatefulSymbolicContext(dynamic_sizes=[, , ], dynamic_strides=[, , ], constraint_sizes=[StrictMinMaxConstraint(warn_only=False, vr=VR[858, 858]), StrictMinMaxConstraint(warn_only=False, vr=VR[6, 6]), StrictMinMaxConstraint(warn_only=False, vr=VR[2, 2])], constraint_strides=[None, None, None], view_base_context=None, tensor_source=LocalSource(local_name='position_sequence', cell_or_freevar=False), shape_env_to_source_to_symbol_cache={}) V1219 16:33:18.588000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:185 in forward (LearnedSimulatorFull.forward) V1219 16:33:18.588000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] data = self.build_graph(particle_type, position_sequence) V1219 16:33:18.591000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST self [] V1219 16:33:18.591000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR build_graph [NNModuleVariable()] V1219 16:33:18.592000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST particle_type [NNModuleVariable()] V1219 16:33:18.592000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST position_sequence [NNModuleVariable(), TensorVariable()] V1219 16:33:18.592000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [NNModuleVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.593000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3099] [0/0] INLINING , inlined according trace_rules.lookup MOD_INLINELIST V1219 16:33:18.593000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1741 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.593000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] forward_call = (self._slow_forward if torch._C._get_tracing_state() else self.forward) V1219 16:33:18.612000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.613000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR _C [PythonModuleVariable()] V1219 16:33:18.613000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR _get_tracing_state [PythonModuleVariable()] V1219 16:33:18.613000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 0 [TorchInGraphFunctionVariable()] V1219 16:33:18.613000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_FALSE 16 [ConstantVariable()] V1219 16:33:18.613000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF self [] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR forward [NNModuleVariable()] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_DEREF forward_call [UserMethodVariable()] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1744 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks or self._forward_pre_hooks V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF self [] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR _backward_hooks [NNModuleVariable()] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF self [] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR _backward_pre_hooks [NNModuleVariable()] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF self [] V1219 16:33:18.614000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR _forward_hooks [NNModuleVariable()] V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF self [] V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR _forward_pre_hooks [NNModuleVariable()] V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1745 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] or _global_backward_pre_hooks or _global_backward_hooks V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL _global_backward_pre_hooks [] V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1744 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks or self._forward_pre_hooks V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1745 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] or _global_backward_pre_hooks or _global_backward_hooks V1219 16:33:18.615000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL _global_backward_hooks [] V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1744 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks or self._forward_pre_hooks V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1746 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] or _global_forward_hooks or _global_forward_pre_hooks): V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL _global_forward_hooks [] V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1744 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks or self._forward_pre_hooks V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1746 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] or _global_forward_hooks or _global_forward_pre_hooks): V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL _global_forward_pre_hooks [] V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1744 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] if not (self._backward_hooks or self._backward_pre_hooks or self._forward_hooks or self._forward_pre_hooks V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_JUMP_IF_TRUE 76 [ConstDictVariable()] V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py:1747 in _call_impl (Module._call_impl) (inline depth: 1) V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] return forward_call(*args, **kwargs) V1219 16:33:18.616000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF forward_call [] V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF args [UserMethodVariable()] V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_MAP 0 [UserMethodVariable(), TupleVariable(length=2)] V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF kwargs [UserMethodVariable(), TupleVariable(length=2), ConstDictVariable()] V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE DICT_MERGE 1 [UserMethodVariable(), TupleVariable(length=2), ConstDictVariable(), ConstDictVariable()] V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_EX 1 [UserMethodVariable(), TupleVariable(length=2), ConstDictVariable()] V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3099] [0/0] INLINING , inlined according trace_rules.lookup inlined by default V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:108 in forward (BuildGraph.forward) (inline depth: 2) V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] graph = preprocess(particle_type, position_sequence, self.metadata) V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL preprocess [] V1219 16:33:18.617000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST particle_type [UserFunctionVariable()] V1219 16:33:18.618000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST position_sequence [UserFunctionVariable(), TensorVariable()] V1219 16:33:18.618000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST self [UserFunctionVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.618000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR metadata [UserFunctionVariable(), TensorVariable(), TensorVariable(), NNModuleVariable()] V1219 16:33:18.618000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 3 [UserFunctionVariable(), TensorVariable(), TensorVariable(), ConstDictVariable()] V1219 16:33:18.618000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3099] [0/0] INLINING , inlined according trace_rules.lookup inlined by default V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:68 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] recent_position = position_seq[:, -1] V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST position_seq [] V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable()] V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), ConstantVariable()] V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TensorVariable(), SliceVariable()] V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [TensorVariable(), SliceVariable(), ConstantVariable()] V1219 16:33:18.619000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.621000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST recent_position [TensorVariable()] V1219 16:33:18.621000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:69 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.621000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] velocity_seq = position_seq[:, 1:] - position_seq[:, :-1] V1219 16:33:18.621000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST position_seq [] V1219 16:33:18.621000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable()] V1219 16:33:18.622000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), ConstantVariable()] V1219 16:33:18.622000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.622000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1 [TensorVariable(), SliceVariable()] V1219 16:33:18.622000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), SliceVariable(), ConstantVariable()] V1219 16:33:18.622000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [TensorVariable(), SliceVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.622000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [TensorVariable(), SliceVariable(), SliceVariable()] V1219 16:33:18.622000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.623000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST position_seq [TensorVariable()] V1219 16:33:18.623000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), TensorVariable()] V1219 16:33:18.623000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.623000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [TensorVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.623000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), TensorVariable(), SliceVariable()] V1219 16:33:18.624000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TensorVariable(), TensorVariable(), SliceVariable(), ConstantVariable()] V1219 16:33:18.624000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [TensorVariable(), TensorVariable(), SliceVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.624000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [TensorVariable(), TensorVariable(), SliceVariable(), SliceVariable()] V1219 16:33:18.624000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TensorVariable(), TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.625000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBTRACT None [TensorVariable(), TensorVariable()] V1219 16:33:18.626000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST velocity_seq [TensorVariable()] V1219 16:33:18.626000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:73 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.626000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_index = find_connectivity(recent_position, metadata["default_connectivity_radius"]) V1219 16:33:18.626000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL find_connectivity [] V1219 16:33:18.626000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST recent_position [UserFunctionVariable()] V1219 16:33:18.626000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST metadata [UserFunctionVariable(), TensorVariable()] V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST default_connectivity_radius [UserFunctionVariable(), TensorVariable(), ConstDictVariable()] V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [UserFunctionVariable(), TensorVariable(), ConstDictVariable(), ConstantVariable()] V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [UserFunctionVariable(), TensorVariable(), LazyVariableTracker()] V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3099] [0/0] INLINING , inlined according trace_rules.lookup inlined by default V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:51 in find_connectivity (find_connectivity) (inline depth: 4) V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] squared_norm = torch.sum(positions*positions, 1) # [N_particles] V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.627000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR sum [PythonModuleVariable()] V1219 16:33:18.628000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST positions [TorchInGraphFunctionVariable()] V1219 16:33:18.628000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST positions [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.628000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_MULTIPLY None [TorchInGraphFunctionVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.629000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1 [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.629000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST squared_norm [TensorVariable()] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:52 in find_connectivity (find_connectivity) (inline depth: 4) V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] squared_norm = torch.reshape(squared_norm, [-1, 1]) # [N_particles, 1] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR reshape [PythonModuleVariable()] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST squared_norm [TorchInGraphFunctionVariable()] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_LIST 2 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.630000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [TorchInGraphFunctionVariable(), TensorVariable(), ListVariable(length=2)] V1219 16:33:18.631000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST squared_norm [TensorVariable()] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:53 in find_connectivity (find_connectivity) (inline depth: 4) V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] distance_tensor = squared_norm - 2*torch.matmul(positions, torch.transpose(positions, 0, 1)) + torch.transpose(squared_norm, 0, 1) # [N_particles, N_particles] Pair-wise square distance matrix V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST squared_norm [] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 2 [TensorVariable()] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [TensorVariable(), ConstantVariable()] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR matmul [TensorVariable(), ConstantVariable(), PythonModuleVariable()] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST positions [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable()] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR transpose [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable(), PythonModuleVariable()] V1219 16:33:18.632000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST positions [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable(), TorchInGraphFunctionVariable()] V1219 16:33:18.633000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 0 [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.633000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1 [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.633000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 3 [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.634000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.635000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_MULTIPLY None [TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.635000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBTRACT None [TensorVariable(), TensorVariable()] V1219 16:33:18.636000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [TensorVariable()] V1219 16:33:18.636000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR transpose [TensorVariable(), PythonModuleVariable()] V1219 16:33:18.636000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST squared_norm [TensorVariable(), TorchInGraphFunctionVariable()] V1219 16:33:18.637000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 0 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.637000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.637000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 3 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.637000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_ADD None [TensorVariable(), TensorVariable()] V1219 16:33:18.638000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST distance_tensor [TensorVariable()] V1219 16:33:18.638000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:56 in find_connectivity (find_connectivity) (inline depth: 4) V1219 16:33:18.638000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_index = torch.nonzero(torch.less_equal(distance_tensor, radius * radius), as_tuple=False) V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR nonzero [PythonModuleVariable()] V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [TorchInGraphFunctionVariable()] V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR less_equal [TorchInGraphFunctionVariable(), PythonModuleVariable()] V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST distance_tensor [TorchInGraphFunctionVariable(), TorchInGraphFunctionVariable()] V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST radius [TorchInGraphFunctionVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST radius [TorchInGraphFunctionVariable(), TorchInGraphFunctionVariable(), TensorVariable(), LazyVariableTracker()] V1219 16:33:18.639000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_MULTIPLY None [TorchInGraphFunctionVariable(), TorchInGraphFunctionVariable(), TensorVariable(), LazyVariableTracker(), LazyVariableTracker()] V1219 16:33:18.640000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [TorchInGraphFunctionVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.641000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST False [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.641000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('as_tuple',) [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.641000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 2 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TupleVariable(length=1)] I1219 16:33:18.642000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:3317] [0/0] create_unbacked_symint u0 [-int_oo, int_oo] at export_gnn_executorch.py:56 in find_connectivity (_subclasses/fake_impls.py:426 in nonzero) V1219 16:33:18.643000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:4734] [0/0] _update_var_to_range u0 = VR[0, 736164] (update) I1219 16:33:18.643000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5481] [0/0] constrain_symbol_range u0 [0, 736164] V1219 16:33:18.659000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.661000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.662000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval Eq(u0, 0) == False [statically known] V1219 16:33:18.663000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] I1219 16:33:18.663000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:604] [0/0] compute_unbacked_bindings [u0] V1219 16:33:18.664000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST edge_index [TensorVariable()] V1219 16:33:18.664000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:58 in find_connectivity (find_connectivity) (inline depth: 4) V1219 16:33:18.664000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] return edge_index.T V1219 16:33:18.664000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_index [] V1219 16:33:18.664000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR T [TensorVariable()] V1219 16:33:18.665000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE RETURN_VALUE None [TensorVariable()] V1219 16:33:18.665000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3164] [0/0] DONE INLINING V1219 16:33:18.665000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST edge_index [TensorVariable()] V1219 16:33:18.665000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:76 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.665000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] boundary = torch.tensor(metadata["bounds"]) V1219 16:33:18.665000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.665000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR tensor [PythonModuleVariable()] V1219 16:33:18.666000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST metadata [TorchInGraphFunctionVariable()] V1219 16:33:18.666000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST bounds [TorchInGraphFunctionVariable(), ConstDictVariable()] V1219 16:33:18.666000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TorchInGraphFunctionVariable(), ConstDictVariable(), ConstantVariable()] V1219 16:33:18.666000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 1 [TorchInGraphFunctionVariable(), LazyVariableTracker()] V1219 16:33:18.667000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST boundary [TensorVariable()] V1219 16:33:18.667000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:77 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.667000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] distance_to_lower_boundary = recent_position - boundary[:, 0] V1219 16:33:18.667000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST recent_position [] V1219 16:33:18.667000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST boundary [TensorVariable()] V1219 16:33:18.667000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), TensorVariable()] V1219 16:33:18.668000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.668000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [TensorVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.668000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 0 [TensorVariable(), TensorVariable(), SliceVariable()] V1219 16:33:18.668000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [TensorVariable(), TensorVariable(), SliceVariable(), ConstantVariable()] V1219 16:33:18.668000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TensorVariable(), TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.669000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBTRACT None [TensorVariable(), TensorVariable()] V1219 16:33:18.670000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST distance_to_lower_boundary [TensorVariable()] V1219 16:33:18.670000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:78 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.670000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] distance_to_upper_boundary = boundary[:, 1] - recent_position V1219 16:33:18.670000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST boundary [] V1219 16:33:18.670000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable()] V1219 16:33:18.670000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [TensorVariable(), ConstantVariable()] V1219 16:33:18.670000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.671000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1 [TensorVariable(), SliceVariable()] V1219 16:33:18.671000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [TensorVariable(), SliceVariable(), ConstantVariable()] V1219 16:33:18.671000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.672000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST recent_position [TensorVariable()] V1219 16:33:18.672000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBTRACT None [TensorVariable(), TensorVariable()] V1219 16:33:18.672000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST distance_to_upper_boundary [TensorVariable()] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:79 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] distance_to_boundary = torch.cat((distance_to_lower_boundary, distance_to_upper_boundary), dim=-1) V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR cat [PythonModuleVariable()] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST distance_to_lower_boundary [TorchInGraphFunctionVariable()] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST distance_to_upper_boundary [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [TorchInGraphFunctionVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TorchInGraphFunctionVariable(), TupleVariable(length=2)] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('dim',) [TorchInGraphFunctionVariable(), TupleVariable(length=2), ConstantVariable()] V1219 16:33:18.673000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 2 [TorchInGraphFunctionVariable(), TupleVariable(length=2), ConstantVariable(), TupleVariable(length=1)] V1219 16:33:18.674000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST distance_to_boundary [TensorVariable()] V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:80 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] distance_to_boundary = torch.clip(distance_to_boundary / metadata["default_connectivity_radius"], -1.0, 1.0) V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR clip [PythonModuleVariable()] V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST distance_to_boundary [TorchInGraphFunctionVariable()] V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST metadata [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST default_connectivity_radius [TorchInGraphFunctionVariable(), TensorVariable(), ConstDictVariable()] V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TorchInGraphFunctionVariable(), TensorVariable(), ConstDictVariable(), ConstantVariable()] V1219 16:33:18.675000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_TRUE_DIVIDE None [TorchInGraphFunctionVariable(), TensorVariable(), LazyVariableTracker()] V1219 16:33:18.676000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1.0 [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.676000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1.0 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.676000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 3 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.678000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST distance_to_boundary [TensorVariable()] V1219 16:33:18.678000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:83 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.678000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] dim = recent_position.size(-1) V1219 16:33:18.678000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST recent_position [] V1219 16:33:18.678000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR size [TensorVariable()] V1219 16:33:18.678000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [GetAttrVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 1 [GetAttrVariable(), ConstantVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST dim [ConstantVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:84 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_displacement = (torch.gather(recent_position, dim=0, index=edge_index[0].unsqueeze(-1).expand(-1, dim)) - V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR gather [PythonModuleVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST recent_position [TorchInGraphFunctionVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 0 [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_index [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 0 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.679000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.680000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR unsqueeze [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.680000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable()] V1219 16:33:18.680000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 1 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable(), ConstantVariable()] V1219 16:33:18.682000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval Eq(u0, 1) == False [statically known] V1219 16:33:18.682000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.683000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval False == False [statically known] V1219 16:33:18.683000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR expand [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.683000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable()] V1219 16:33:18.683000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST dim [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable(), ConstantVariable()] V1219 16:33:18.683000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.685000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval Eq(u0, -1) == False [statically known] V1219 16:33:18.686000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.686000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.687000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('dim', 'index') [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.687000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 3 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.688000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval Eq(2*u0, 0) == False [statically known] V1219 16:33:18.689000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.689000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:85 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.689000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] torch.gather(recent_position, dim=0, index=edge_index[1].unsqueeze(-1).expand(-1, dim))) V1219 16:33:18.689000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [TensorVariable()] V1219 16:33:18.690000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR gather [TensorVariable(), PythonModuleVariable()] V1219 16:33:18.690000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST recent_position [TensorVariable(), TorchInGraphFunctionVariable()] V1219 16:33:18.690000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 0 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.690000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_index [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.690000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 1 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.690000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.691000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR unsqueeze [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.691000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable()] V1219 16:33:18.691000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 1 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable(), ConstantVariable()] V1219 16:33:18.691000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.692000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR expand [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.692000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable()] V1219 16:33:18.692000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST dim [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable(), ConstantVariable()] V1219 16:33:18.692000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), GetAttrVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.693000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.693000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.694000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('dim', 'index') [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.694000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 3 [TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.694000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.695000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:84 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.695000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_displacement = (torch.gather(recent_position, dim=0, index=edge_index[0].unsqueeze(-1).expand(-1, dim)) - V1219 16:33:18.695000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBTRACT None [TensorVariable(), TensorVariable()] V1219 16:33:18.695000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.696000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.696000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval True == True [statically known] V1219 16:33:18.696000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST edge_displacement [TensorVariable()] V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:86 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_displacement /= metadata["default_connectivity_radius"] V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_displacement [] V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST metadata [TensorVariable()] V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST default_connectivity_radius [TensorVariable(), ConstDictVariable()] V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [TensorVariable(), ConstDictVariable(), ConstantVariable()] V1219 16:33:18.697000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE INPLACE_TRUE_DIVIDE None [TensorVariable(), LazyVariableTracker()] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST edge_displacement [TensorVariable()] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:87 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_distance = torch.norm(edge_displacement, dim=-1, keepdim=True) V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR norm [PythonModuleVariable()] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_displacement [TorchInGraphFunctionVariable()] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST True [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('dim', 'keepdim') [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.698000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 3 [TorchInGraphFunctionVariable(), TensorVariable(), ConstantVariable(), ConstantVariable(), TupleVariable(length=2)] V1219 16:33:18.700000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval u0 < 0 == False [statically known] V1219 16:33:18.700000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.700000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.701000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.701000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.703000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.704000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.704000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.705000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval True == True [statically known] V1219 16:33:18.706000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5201] [0/0] eval Ne(u0, 1) == True [statically known] V1219 16:33:18.707000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.707000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.708000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_FAST edge_distance [TensorVariable()] V1219 16:33:18.708000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:90 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.708000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] graph = pyg.data.Data( V1219 16:33:18.708000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL pyg [] V1219 16:33:18.708000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR data [PythonModuleVariable()] V1219 16:33:18.708000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR Data [PythonModuleVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:91 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] x=particle_type, V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST particle_type [UserDefinedClassVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:92 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_index=edge_index, V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_index [UserDefinedClassVariable(), TensorVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:93 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] edge_attr=torch.cat((edge_displacement, edge_distance), dim=-1), V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [UserDefinedClassVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR cat [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), PythonModuleVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_displacement [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TorchInGraphFunctionVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_distance [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.709000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TupleVariable(length=2)] V1219 16:33:18.710000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('dim',) [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TupleVariable(length=2), ConstantVariable()] V1219 16:33:18.710000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 2 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TorchInGraphFunctionVariable(), TupleVariable(length=2), ConstantVariable(), TupleVariable(length=1)] V1219 16:33:18.710000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.710000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert True == True [statically known] V1219 16:33:18.711000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.712000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.713000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.714000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.714000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:5358] [0/0] runtime_assert u0 >= 0 == True [statically known] V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:94 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] y=None, # Ground truth for training V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:95 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] pos=torch.cat((velocity_seq.reshape(velocity_seq.size(0), -1), distance_to_boundary), dim=-1), V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL torch [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR cat [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), PythonModuleVariable()] V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST velocity_seq [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable()] V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR reshape [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST velocity_seq [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), GetAttrVariable()] V1219 16:33:18.715000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR size [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), GetAttrVariable(), TensorVariable()] V1219 16:33:18.716000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST 0 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), GetAttrVariable(), GetAttrVariable()] V1219 16:33:18.716000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 1 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), GetAttrVariable(), GetAttrVariable(), ConstantVariable()] V1219 16:33:18.716000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), GetAttrVariable(), ConstantVariable()] V1219 16:33:18.716000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), GetAttrVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.717000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST distance_to_boundary [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable()] V1219 16:33:18.717000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.717000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TupleVariable(length=2)] V1219 16:33:18.717000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('dim',) [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TupleVariable(length=2), ConstantVariable()] V1219 16:33:18.717000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 2 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TorchInGraphFunctionVariable(), TupleVariable(length=2), ConstantVariable(), TupleVariable(length=1)] V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:96 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] recent_position=recent_position, V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST recent_position [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable()] V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:97 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] recent_velocity=velocity_seq[:, -1] V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST velocity_seq [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST None [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable()] V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_SLICE 2 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), ConstantVariable()] V1219 16:33:18.718000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST -1 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable(), SliceVariable()] V1219 16:33:18.719000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BUILD_TUPLE 2 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable(), SliceVariable(), ConstantVariable()] V1219 16:33:18.719000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE BINARY_SUBSCR None [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable(), TupleVariable(length=2)] V1219 16:33:18.720000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py:90 in preprocess (preprocess) (inline depth: 3) V1219 16:33:18.720000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] graph = pyg.data.Data( V1219 16:33:18.720000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('x', 'edge_index', 'edge_attr', 'y', 'pos', 'recent_position', 'recent_velocity') [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable()] V1219 16:33:18.720000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 7 [UserDefinedClassVariable(), TensorVariable(), TensorVariable(), TensorVariable(), ConstantVariable(), TensorVariable(), TensorVariable(), TensorVariable(), TupleVariable(length=7)] V1219 16:33:18.721000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3099] [0/0] INLINING , inlined according trace_rules.lookup inlined by default V1219 16:33:18.721000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch_geometric/data/data.py:530 in __init__ (Data.__init__) (inline depth: 4) V1219 16:33:18.721000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] super().__init__(tensor_attr_cls=DataTensorAttr) V1219 16:33:18.732000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL super [] V1219 16:33:18.733000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF __class__ [BuiltinVariable()] V1219 16:33:18.733000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST self [BuiltinVariable(), UserDefinedClassVariable()] V1219 16:33:18.733000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [BuiltinVariable(), UserDefinedClassVariable(), UserDefinedObjectVariable(Data)] V1219 16:33:18.733000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR __init__ [SuperVariable()] V1219 16:33:18.733000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL DataTensorAttr [GetAttrVariable()] V1219 16:33:18.733000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST ('tensor_attr_cls',) [GetAttrVariable(), UserDefinedClassVariable()] V1219 16:33:18.733000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION_KW 1 [GetAttrVariable(), UserDefinedClassVariable(), TupleVariable(length=1)] V1219 16:33:18.734000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3099] [0/0] INLINING , inlined according trace_rules.lookup inlined by default V1219 16:33:18.734000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch_geometric/data/feature_store.py:278 in __init__ (FeatureStore.__init__) (inline depth: 5) V1219 16:33:18.734000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] super().__init__() V1219 16:33:18.737000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL super [] V1219 16:33:18.737000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF __class__ [BuiltinVariable()] V1219 16:33:18.737000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST self [BuiltinVariable(), UserDefinedClassVariable()] V1219 16:33:18.737000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [BuiltinVariable(), UserDefinedClassVariable(), UserDefinedObjectVariable(Data)] V1219 16:33:18.737000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR __init__ [SuperVariable()] V1219 16:33:18.738000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 0 [GetAttrVariable()] V1219 16:33:18.738000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3099] [0/0] INLINING , inlined according trace_rules.lookup inlined by default V1219 16:33:18.738000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch_geometric/data/graph_store.py:110 in __init__ (GraphStore.__init__) (inline depth: 6) V1219 16:33:18.738000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] super().__init__() V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL super [] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_DEREF __class__ [BuiltinVariable()] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST self [BuiltinVariable(), UserDefinedClassVariable()] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 2 [BuiltinVariable(), UserDefinedClassVariable(), UserDefinedObjectVariable(Data)] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR __init__ [SuperVariable()] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE CALL_FUNCTION 0 [GetAttrVariable()] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE POP_TOP None [LambdaVariable()] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] TRACE starts_line /home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch_geometric/data/graph_store.py:111 in __init__ (GraphStore.__init__) (inline depth: 6) V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:865] [0/0] [__trace_source] self.__dict__['_edge_attr_cls'] = edge_attr_cls or EdgeAttr V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST edge_attr_cls [] V1219 16:33:18.741000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE JUMP_IF_TRUE_OR_POP 16 [ConstantVariable()] V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_GLOBAL EdgeAttr [] V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_FAST self [UserDefinedClassVariable()] V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_ATTR __dict__ [UserDefinedClassVariable(), UserDefinedObjectVariable(Data)] V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE LOAD_CONST _edge_attr_cls [UserDefinedClassVariable(), GetAttrVariable()] V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:888] [0/0] [__trace_bytecode] TRACE STORE_SUBSCR None [UserDefinedClassVariable(), GetAttrVariable(), ConstantVariable()] V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:904] [0/0] empty checkpoint V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3153] [0/0] FAILED INLINING V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:904] [0/0] empty checkpoint V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3153] [0/0] FAILED INLINING V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:904] [0/0] empty checkpoint V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3153] [0/0] FAILED INLINING V1219 16:33:18.742000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:904] [0/0] empty checkpoint V1219 16:33:18.743000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3153] [0/0] FAILED INLINING V1219 16:33:18.743000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:904] [0/0] empty checkpoint V1219 16:33:18.743000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3153] [0/0] FAILED INLINING V1219 16:33:18.743000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:904] [0/0] empty checkpoint V1219 16:33:18.743000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:3153] [0/0] FAILED INLINING V1219 16:33:18.743000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py:904] [0/0] empty checkpoint PyTorch has version 2.5.0+cu124 with cuda 12.4 -1 0 -1 0 Traceback (most recent call last): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py", line 227, in aten_dialect = export(simulator, example_inputs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/__init__.py", line 270, in export return _export( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 1017, in wrapper raise e File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 990, in wrapper ep = fn(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/exported_program.py", line 114, in wrapper return fn(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 1880, in _export export_artifact = export_func( # type: ignore[operator] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 1224, in _strict_export return _strict_export_lower_to_aten_ir( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 1252, in _strict_export_lower_to_aten_ir gm_torch_level = _export_to_torch_ir( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/export/_trace.py", line 560, in _export_to_torch_ir gm_torch_level, _ = torch._dynamo.export( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 1432, in inner result_traced = opt_f(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 465, in _fn return fn(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl return self._call_impl(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 1269, in __call__ return self._torchdynamo_orig_callable( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 526, in __call__ return _compile( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 924, in _compile guarded_code = compile_inner(code, one_graph, hooks, transform) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 666, in compile_inner return _compile_inner(code, one_graph, hooks, transform) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_utils_internal.py", line 87, in wrapper_function return function(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 699, in _compile_inner out_code = transform_code_object(code, transform) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/bytecode_transformation.py", line 1322, in transform_code_object transformations(instructions, code_options) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 219, in _fn return fn(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 634, in transform tracer.run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 2796, in run super().run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 983, in run while self.step(): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 895, in step self.dispatch_table[inst.opcode](self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 582, in wrapper return inner_fn(self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 1602, in CALL_FUNCTION self.call_function(fn, args, {}) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 830, in call_function self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/nn_module.py", line 442, in call_function return tx.inline_user_function_return( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 836, in inline_user_function_return return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3011, in inline_call return cls.inline_call_(parent, func, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3139, in inline_call_ tracer.run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 983, in run while self.step(): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 895, in step self.dispatch_table[inst.opcode](self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 582, in wrapper return inner_fn(self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 1680, in CALL_FUNCTION_EX self.call_function(fn, argsvars.items, kwargsvars) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 830, in call_function self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 385, in call_function return super().call_function(tx, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 324, in call_function return super().call_function(tx, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 111, in call_function return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 836, in inline_user_function_return return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3011, in inline_call return cls.inline_call_(parent, func, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3139, in inline_call_ tracer.run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 983, in run while self.step(): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 895, in step self.dispatch_table[inst.opcode](self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 582, in wrapper return inner_fn(self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 1602, in CALL_FUNCTION self.call_function(fn, args, {}) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 830, in call_function self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 324, in call_function return super().call_function(tx, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 111, in call_function return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 836, in inline_user_function_return return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3011, in inline_call return cls.inline_call_(parent, func, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3139, in inline_call_ tracer.run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 983, in run while self.step(): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 895, in step self.dispatch_table[inst.opcode](self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 582, in wrapper return inner_fn(self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 1692, in CALL_FUNCTION_KW self.call_function(fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 830, in call_function self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/user_defined.py", line 496, in call_function var.call_method(tx, "__init__", args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/user_defined.py", line 788, in call_method return UserMethodVariable(method, self, source=source).call_function( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 385, in call_function return super().call_function(tx, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 324, in call_function return super().call_function(tx, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 111, in call_function return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 836, in inline_user_function_return return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3011, in inline_call return cls.inline_call_(parent, func, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3139, in inline_call_ tracer.run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 983, in run while self.step(): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 895, in step self.dispatch_table[inst.opcode](self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 582, in wrapper return inner_fn(self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 1692, in CALL_FUNCTION_KW self.call_function(fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 830, in call_function self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/misc.py", line 1024, in call_function return self.obj.call_method(tx, self.name, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/misc.py", line 195, in call_method ).call_function(tx, [self.objvar] + args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 324, in call_function return super().call_function(tx, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 111, in call_function return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 836, in inline_user_function_return return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3011, in inline_call return cls.inline_call_(parent, func, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3139, in inline_call_ tracer.run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 983, in run while self.step(): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 895, in step self.dispatch_table[inst.opcode](self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 582, in wrapper return inner_fn(self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 1602, in CALL_FUNCTION self.call_function(fn, args, {}) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 830, in call_function self.push(fn.call_function(self, args, kwargs)) # type: ignore[arg-type] File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/misc.py", line 1024, in call_function return self.obj.call_method(tx, self.name, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/misc.py", line 195, in call_method ).call_function(tx, [self.objvar] + args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 324, in call_function return super().call_function(tx, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/functions.py", line 111, in call_function return tx.inline_user_function_return(self, [*self.self_args(), *args], kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 836, in inline_user_function_return return InliningInstructionTranslator.inline_call(self, fn, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3011, in inline_call return cls.inline_call_(parent, func, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 3139, in inline_call_ tracer.run() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 983, in run while self.step(): File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 895, in step self.dispatch_table[inst.opcode](self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 582, in wrapper return inner_fn(self, inst) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 1803, in STORE_SUBSCR result = obj.call_method(self, "__setitem__", [key, val], {}) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/misc.py", line 1082, in call_method return super().call_method(tx, name, args, kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/variables/base.py", line 343, in call_method unimplemented(f"call_method {self} {name} {args} {kwargs}") File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/_dynamo/exc.py", line 297, in unimplemented raise Unsupported(msg, case_name=case_name) torch._dynamo.exc.Unsupported: call_method GetAttrVariable(UserDefinedObjectVariable(Data), __dict__) __setitem__ [ConstantVariable(), UserDefinedClassVariable()] {} from user code: File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py", line 185, in forward data = self.build_graph(particle_type, position_sequence) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl return forward_call(*args, **kwargs) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py", line 108, in forward graph = preprocess(particle_type, position_sequence, self.metadata) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/export_gnn_executorch.py", line 90, in preprocess graph = pyg.data.Data( File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch_geometric/data/data.py", line 530, in __init__ super().__init__(tensor_attr_cls=DataTensorAttr) File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch_geometric/data/feature_store.py", line 278, in __init__ super().__init__() File "/home/sicli01/Projects/FluidML/gnn-physics-pytorch/gnn_env/lib/python3.10/site-packages/torch_geometric/data/graph_store.py", line 111, in __init__ self.__dict__['_edge_attr_cls'] = edge_attr_cls or EdgeAttr I1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:399] TorchDynamo compilation metrics: I1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:399] Function Runtimes (s) I1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:399] ---------------------- -------------- I1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:399] _compile.compile_inner 0 V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats constrain_symbol_range: CacheInfo(hits=18, misses=2, maxsize=None, currsize=2) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats evaluate_expr: CacheInfo(hits=39, misses=9, maxsize=256, currsize=9) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _simplify_floor_div: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _maybe_guard_rel: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _find: CacheInfo(hits=20, misses=1, maxsize=None, currsize=1) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats has_hint: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats size_hint: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats simplify: CacheInfo(hits=4, misses=9, maxsize=None, currsize=9) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _update_divisible: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:18.755000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats replace: CacheInfo(hits=1208, misses=30, maxsize=None, currsize=30) V1219 16:33:18.756000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _maybe_evaluate_static: CacheInfo(hits=49, misses=13, maxsize=None, currsize=13) V1219 16:33:18.756000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats get_implications: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:18.756000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats get_axioms: CacheInfo(hits=10, misses=3, maxsize=None, currsize=3) V1219 16:33:18.756000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats safe_expand: CacheInfo(hits=182, misses=30, maxsize=256, currsize=30) V1219 16:33:18.756000 2552752 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats uninteresting_files: CacheInfo(hits=4, misses=1, maxsize=None, currsize=1) I1219 16:33:19.704000 2552920 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:399] TorchDynamo compilation metrics: I1219 16:33:19.704000 2552920 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:399] Function Runtimes (s) I1219 16:33:19.704000 2552920 gnn_env/lib/python3.10/site-packages/torch/_dynamo/utils.py:399] ---------- -------------- V1219 16:33:19.704000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats constrain_symbol_range: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.704000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats evaluate_expr: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:19.704000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _simplify_floor_div: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.704000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _maybe_guard_rel: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _find: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats has_hint: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats size_hint: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats simplify: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _update_divisible: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats replace: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats _maybe_evaluate_static: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats get_implications: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats get_axioms: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats safe_expand: CacheInfo(hits=0, misses=0, maxsize=256, currsize=0) V1219 16:33:19.705000 2552920 gnn_env/lib/python3.10/site-packages/torch/fx/experimental/symbolic_shapes.py:122] lru_cache_stats uninteresting_files: CacheInfo(hits=0, misses=0, maxsize=None, currsize=0)