Oleg Khabinov 4b18ab869f [torch.export] Support is_compiling() flag for non-strict mode (#119602) ...

Summary: In non-strict mode of torch.export() we didn't set those `is_compiling()` to `True` which is needed by some models.

Test Plan: Unit tests and manual testing.

Differential Revision: D53624452

Pull Request resolved: https://github.com/pytorch/pytorch/pull/119602
Approved by: https://github.com/suo

2024-02-29 05:52:51 +00:00

..

__init__.py

[torch.export] Support is_compiling() flag for non-strict mode (#119602)

2024-02-29 05:52:51 +00:00