sglang_v0.5.2/pytorch_2.8.0/test/jit/myfunction_a.py

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