Skip to content

Instantly share code, notes, and snippets.

@davidberard98
Last active March 15, 2024 00:00
Show Gist options
  • Save davidberard98/0fc1a4849f2ac6eb4d2e09fb42bb7cd5 to your computer and use it in GitHub Desktop.
Save davidberard98/0fc1a4849f2ac6eb4d2e09fb42bb7cd5 to your computer and use it in GitHub Desktop.
E
======================================================================
ERROR: test_torch_function_call_to_size_within_aot_autograd_graph (__main__.TestNestedTensor.test_torch_function_call_to_size_within_aot_autograd_graph)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/data/users/dberard/pytorch/torch/testing/_internal/common_utils.py", line 2739, in wrapper
method(*args, **kwargs)
File "/data/users/dberard/pytorch/test/dynamo/test_subclasses.py", line 1403, in test_torch_function_call_to_size_within_aot_autograd_graph
compiled_fn(x, y)
File "/data/users/dberard/pytorch/torch/_dynamo/eval_frame.py", line 451, in _fn
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/convert_frame.py", line 921, in catch_errors
return callback(frame, cache_entry, hooks, frame_state, skip=1)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/convert_frame.py", line 400, in _convert_frame_assert
return _compile(
^^^^^^^^^
File "/home/dberard/.conda/envs/pytorch311/lib/python3.11/contextlib.py", line 81, in inner
return func(*args, **kwds)
^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/convert_frame.py", line 676, in _compile
guarded_code = compile_inner(code, one_graph, hooks, transform)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/utils.py", line 262, in time_wrapper
r = func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/convert_frame.py", line 535, in compile_inner
out_code = transform_code_object(code, transform)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/bytecode_transformation.py", line 1036, in transform_code_object
transformations(instructions, code_options)
File "/data/users/dberard/pytorch/torch/_dynamo/convert_frame.py", line 165, in _fn
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/convert_frame.py", line 500, in transform
tracer.run()
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 2149, in run
super().run()
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 810, in run
and self.step()
^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 773, in step
getattr(self, inst.opname)(inst)
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 1311, in LOAD_ATTR
result = BuiltinVariable(getattr).call_function(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/variables/builtin.py", line 687, in call_function
result = handler(tx, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/variables/builtin.py", line 1277, in call_getattr
return obj.var_getattr(tx, name)
^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/variables/torch_function.py", line 218, in var_getattr
return self.call_torch_function(
^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/variables/torch_function.py", line 229, in call_torch_function
return call_torch_function(
^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/variables/torch_function.py", line 94, in call_torch_function
return tx.inline_user_function_return(torch_function_var, tf_args, {})
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 680, in inline_user_function_return
return InliningInstructionTranslator.inline_call(self, fn, args, kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 2285, in inline_call
return cls.inline_call_(parent, func, args, kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 2329, in inline_call_
result = InliningInstructionTranslator.check_inlineable(func)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/data/users/dberard/pytorch/torch/_dynamo/symbolic_convert.py", line 2313, in check_inlineable
unimplemented(
File "/data/users/dberard/pytorch/torch/_dynamo/exc.py", line 190, in unimplemented
raise Unsupported(msg)
torch._dynamo.exc.Unsupported: call torch._dynamo.disable() wrapped function <function TestNestedTensor.test_torch_function_call_to_size_within_aot_autograd_graph.<locals>.TestTensor.__torch_function__ at 0x7f24bc369620>
from user code:
File "/data/users/dberard/pytorch/test/dynamo/test_subclasses.py", line 1391, in fn
z = x.shape[0] * y.shape[0]
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
To execute this test, run the following from the base repo dir:
python test/dynamo/test_subclasses.py -k test_torch_function_call_to_size_within_aot_autograd_graph
This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0
----------------------------------------------------------------------
Ran 1 test in 0.128s
FAILED (errors=1)
inline_call [('call torch._dynamo.disable() wrapped function <function TestNestedTensor.test_torch_function_call_to_size_within_aot_autograd_graph.<locals>.TestTensor.__torch_function__ at 0x7f24bc369620>', 1)]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment