Check whether the module was compiled with torch.compile()
(module)
| 83 | |
| 84 | |
| 85 | def is_compiled_module(module) -> bool: |
| 86 | """Check whether the module was compiled with torch.compile()""" |
| 87 | if is_torch_version("<", "2.0.0") or not hasattr(torch, "_dynamo"): |
| 88 | return False |
| 89 | return isinstance(module, torch._dynamo.eval_frame.OptimizedModule) |
| 90 | |
| 91 | |
| 92 | def fourier_filter(x_in: "torch.Tensor", threshold: int, scale: int) -> "torch.Tensor": |