xref: /aosp_15_r20/external/pytorch/test/jit/myfunction_a.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1"""
2Helper function used in test_decorator.py. We define it in a
3separate file on purpose to test that the names in different modules
4are resolved correctly.
5"""
6
7from jit.mydecorator import my_decorator
8from jit.myfunction_b import my_function_b
9
10
11@my_decorator
12def my_function_a(x: float) -> float:
13    return my_function_b(x) + 1
14