['isonet_maps/J331_007_volume_map_half_A_data', 'isonet_maps/J331_007_volume_map_half_B_data']
0%| | 0/125 [00:00<?, ?batch/s][rank3]:[2024-04-18 14:34:22,274] [0/0] torch._dynamo.variables.torch: [WARNING] Profiler function <class 'torch.autograd.profiler.record_function'> will be ignored
[rank0]:[2024-04-18 14:34:22,618] [0/0] torch._dynamo.variables.torch: [WARNING] Profiler function <class 'torch.autograd.profiler.record_function'> will be ignored
[rank2]:[2024-04-18 14:34:22,899] [0/0] torch._dynamo.variables.torch: [WARNING] Profiler function <class 'torch.autograd.profiler.record_function'> will be ignored
[rank1]:[2024-04-18 14:34:23,284] [0/0] torch._dynamo.variables.torch: [WARNING] Profiler function <class 'torch.autograd.profiler.record_function'> will be ignored
/tmp/tmpr6j127vt/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmp1l43fx85/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmprczwac99/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmpr6j127vt/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpu9tb6e5m/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmp1l43fx85/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpr6j127vt/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmprczwac99/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp1l43fx85/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmprczwac99/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmpr6j127vt/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmp1l43fx85/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmprczwac99/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmpu9tb6e5m/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpr6j127vt/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp1l43fx85/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpu9tb6e5m/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmprczwac99/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpu9tb6e5m/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmpu9tb6e5m/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp3isx_ban/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmpv0o0ry1q/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmp3isx_ban/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp3isx_ban/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmp3isx_ban/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmpv0o0ry1q/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpv0o0ry1q/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmp3isx_ban/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpv0o0ry1q/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmpv0o0ry1q/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp__1zeucj/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmppyr76ww_/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmp__1zeucj/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmppyr76ww_/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp__1zeucj/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmppyr76ww_/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmp__1zeucj/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmppyr76ww_/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmp__1zeucj/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmppyr76ww_/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp24x5boe3/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmp7fkgcdz3/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmp24x5boe3/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp24x5boe3/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmp7fkgcdz3/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp7fkgcdz3/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmp24x5boe3/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmp7fkgcdz3/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmp24x5boe3/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmp7fkgcdz3/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpb0c_5uek/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmpb0c_5uek/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpb0c_5uek/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmpb0c_5uek/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmpb0c_5uek/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpxdg_2cjx/main.c: In function \u2018list_to_cuuint64_array\u2019:
/tmp/tmpxdg_2cjx/main.c:354:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
/tmp/tmpxdg_2cjx/main.c:354:3: note: use option -std=c99 or -std=gnu99 to compile your code
/tmp/tmpxdg_2cjx/main.c: In function \u2018list_to_cuuint32_array\u2019:
/tmp/tmpxdg_2cjx/main.c:365:3: error: \u2018for\u2019 loop initial declarations are only allowed in C99 mode
for (Py_ssize_t i = 0; i < len; i++) {
^
0%| | 0/125 [00:34<?, ?batch/s]
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/multiprocessing/spawn.py", line 158, in join
raise ProcessRaisedException(msg, error_index, failed_process.pid)
torch.multiprocessing.spawn.ProcessRaisedException:
-- Process 2 terminated with the following error:
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/multiprocessing/spawn.py", line 68, in _wrap
fn(i, *args)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/spIsoNet/models/network_n2n.py", line 116, in ddp_train
preds = model(x1)# + noise.cuda())
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 489, in _fn
return fn(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/external_utils.py", line 17, in inner
return fn(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/parallel/distributed.py", line 1523, in forward
else self._run_ddp_forward(*inputs, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/parallel/distributed.py", line 1359, in _run_ddp_forward
return self.module(*inputs, **kwargs) # type: ignore[index]
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
return forward_call(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/spIsoNet/models/unet.py", line 97, in forward
x, down_sampling_features = self.encoder(x)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/spIsoNet/models/unet.py", line 98, in resume_in_forward
x = self.decoder(x, down_sampling_features)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/eval_frame.py", line 652, in catch_errors
return hijacked_callback(frame, cache_entry, hooks, frame_state)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 727, in _convert_frame
result = inner_convert(frame, cache_entry, hooks, frame_state)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 383, in _convert_frame_assert
compiled_product = _compile(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 646, in _compile
guarded_code = compile_inner(code, one_graph, hooks, transform)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/utils.py", line 244, in time_wrapper
r = func(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 562, in compile_inner
out_code = transform_code_object(code, transform)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/bytecode_transformation.py", line 1033, in transform_code_object
transformations(instructions, code_options)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 151, in _fn
return fn(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/convert_frame.py", line 527, in transform
tracer.run()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 2128, in run
super().run()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 818, in run
and self.step()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 781, in step
getattr(self, inst.opname)(inst)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/symbolic_convert.py", line 2243, in RETURN_VALUE
self.output.compile_subgraph(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/output_graph.py", line 919, in compile_subgraph
self.compile_and_call_fx_graph(tx, list(reversed(stack_values)), root)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/contextlib.py", line 79, in inner
return func(*args, **kwds)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/output_graph.py", line 1087, in compile_and_call_fx_graph
compiled_fn = self.call_user_compiler(gm)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/utils.py", line 244, in time_wrapper
r = func(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/output_graph.py", line 1159, in call_user_compiler
raise BackendCompilerFailed(self.compiler_fn, e).with_traceback(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/output_graph.py", line 1140, in call_user_compiler
compiled_fn = compiler_fn(gm, self.example_inputs())
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/backends/distributed.py", line 312, in compile_fn
return self.backend_compile_fn(gm, example_inputs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/repro/after_dynamo.py", line 117, in debug_wrapper
compiled_gm = compiler_fn(gm, example_inputs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/__init__.py", line 1668, in __call__
return compile_fx(model_, inputs_, config_patches=self.config)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/compile_fx.py", line 1168, in compile_fx
return aot_autograd(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/backends/common.py", line 55, in compiler_fn
cg = aot_module_simplified(gm, example_inputs, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_functorch/aot_autograd.py", line 887, in aot_module_simplified
compiled_fn = create_aot_dispatcher_function(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/utils.py", line 244, in time_wrapper
r = func(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_functorch/aot_autograd.py", line 600, in create_aot_dispatcher_function
compiled_fn = compiler_fn(flat_fn, fake_flat_args, aot_config, fw_metadata=fw_metadata)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/runtime_wrappers.py", line 425, in aot_wrapper_dedupe
return compiler_fn(flat_fn, leaf_flat_args, aot_config, fw_metadata=fw_metadata)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/runtime_wrappers.py", line 630, in aot_wrapper_synthetic_base
return compiler_fn(flat_fn, flat_args, aot_config, fw_metadata=fw_metadata)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_functorch/_aot_autograd/jit_compile_runtime_wrappers.py", line 295, in aot_dispatch_autograd
compiled_fw_func = aot_config.fw_compiler(fw_module, adjusted_flat_args)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/utils.py", line 244, in time_wrapper
r = func(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/compile_fx.py", line 1100, in fw_compiler_base
return inner_compile(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/repro/after_aot.py", line 83, in debug_wrapper
inner_compiled_fn = compiler_fn(gm, example_inputs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/debug.py", line 305, in inner
return fn(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/contextlib.py", line 79, in inner
return func(*args, **kwds)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/compile_fx.py", line 320, in compile_fx_inner
compiled_graph = fx_codegen_and_compile(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/compile_fx.py", line 550, in fx_codegen_and_compile
compiled_fn = graph.compile_to_fn()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/graph.py", line 1116, in compile_to_fn
return self.compile_to_module().call
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_dynamo/utils.py", line 244, in time_wrapper
r = func(*args, **kwargs)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/graph.py", line 1070, in compile_to_module
mod = PyCodeCache.load_by_key_path(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/codecache.py", line 1892, in load_by_key_path
exec(code, mod.__dict__, mod.__dict__)
File "/tmp/torchinductor_spetrovi/u6/cu6kxoskl3txa5ptho7vjkz6zh7csuhwxngfjk3opdwi3ab6wbhv.py", line 67, in <module>
async_compile.wait(globals())
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/codecache.py", line 2486, in wait
scope[key] = result.result()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/codecache.py", line 2330, in result
kernel = self.kernel = _load_kernel(self.kernel_name, self.source_code)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/codecache.py", line 2306, in _load_kernel
kernel.precompile()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/triton_heuristics.py", line 188, in precompile
compiled_binary, launcher = self._precompile_config(
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/torch/_inductor/triton_heuristics.py", line 308, in _precompile_config
binary._init_handles()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/compiler/compiler.py", line 670, in _init_handles
bin_path = {driver.HIP: "hsaco_path", driver.CUDA: "cubin"}[driver.backend]
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/runtime/driver.py", line 157, in __getattr__
self._initialize_obj()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/runtime/driver.py", line 154, in _initialize_obj
self._obj = self._init_fn()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/runtime/driver.py", line 187, in initialize_driver
return CudaDriver()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/runtime/driver.py", line 77, in __init__
self.utils = CudaUtils()
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/runtime/driver.py", line 47, in __init__
so = _build("cuda_utils", src_path, tmpdir)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/common/build.py", line 106, in _build
ret = subprocess.check_call(cc_cmd)
File "/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/subprocess.py", line 369, in check_call
raise CalledProcessError(retcode, cmd)
torch._dynamo.exc.BackendCompilerFailed: backend='compile_fn' raised:
CalledProcessError: Command '['/usr/bin/gcc', '/tmp/tmp24x5boe3/main.c', '-O3', '-I/home/spetrovi/miniconda3/envs/spisonet/lib/python3.10/site-packages/triton/common/../third_party/cuda/include', '-I/home/spetrovi/miniconda3/envs/spisonet/include/python3.10', '-I/tmp/tmp24x5boe3', '-shared', '-fPIC', '-lcuda', '-o', '/tmp/tmp24x5boe3/
cuda_utils.cpython-310-x86_64-linux-gnu.so', '-L/lib64', '-L/lib', '-L/lib64', '-L/lib']' returned non-zero exit status 1.
Set TORCH_LOGS="+dynamo" and TORCHDYNAMO_VERBOSE=1 for more information
You can suppress this exception and fall back to eager by setting:
import torch._dynamo
torch._dynamo.config.suppress_errors = True