fix failures- cpu offloading casues device mismatch L2_dynamo_compile…#4089
fix failures- cpu offloading casues device mismatch L2_dynamo_compile…#4089
Conversation
github-actions
bot
added
component: tests
| f"Conversion of module {gm} not currently fully supported or convertible!", | ||
| exc_info=True, | ||
| ) | ||
| raise |
There was a problem hiding this comment.
If we are throwing exceptions, provide an actual error type and explanation
There was a problem hiding this comment.
The bare raise should be able to catch the same exception object or the UnsupportedOperatorException. Do we need another exception message. That would be code repeat. But can add.
| exp_program, | ||
| inputs=[input_tensor], | ||
| enabled_precisions={torch.float8_e4m3fn}, | ||
| use_explicit_typing=False, |
There was a problem hiding this comment.
This is deprecated, this test should be ported to explicit typing
apbose
force-pushed
the
abose/CI_error_fix
branch
2 times, most recently
from
775cd03 to
e33ca90
Compare
|
@narendasan had discussion with @zewenli98. Should we remove this https://github.com/pytorch/TensorRT/blob/main/py/torch_tensorrt/dynamo/_compiler.py#L641-648 from the code, since explicit typing means that we should not provide precision now. |
apbose
force-pushed
the
abose/CI_error_fix
branch
2 times, most recently
from
8ceffc8 to
46e91c3
Compare
… tests, set explicit typing false for precisions test L2_torch_tensorrt_py_tests, and add raise exceptions for error logging
apbose
force-pushed
the
abose/CI_error_fix
branch
from
46e91c3 to
1d8cd02
Compare
2 participants
No description provided.