mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
dccc1ca839
Summary: This fixes a case left incomplete by https://github.com/pytorch/pytorch/pull/106229 The object is using __prepare_scriptable__ correctly inside of torch.jit.script() but the clousre that is obtained below is using the non-prepared version. This causes issues when the prepared and non-prepared versions are in different python modules. Test Plan: ``` buck2 run mode/opt caffe2/test:jit -- -r test_decorator ``` Differential Revision: D54308741 Re-exporting, as #120806 #121307 were not properly merged. Co-authored-by: Daniel Herrera <dherrera@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/121553 Approved by: https://github.com/huydhn, https://github.com/seemethere
14 lines
333 B
Python
14 lines
333 B
Python
"""
|
|
Helper function used in test_decorator.py. We define it in a
|
|
separate file on purpose to test that the names in different modules
|
|
are resolved correctly.
|
|
"""
|
|
|
|
from jit.mydecorator import my_decorator
|
|
from jit.myfunction_b import my_function_b
|
|
|
|
|
|
@my_decorator
|
|
def my_function_a(x: float) -> float:
|
|
return my_function_b(x) + 1
|