inductor.py 267 B

123456789
  1. from torch._dynamo import register_backend
  2. @register_backend
  3. def inductor(*args, **kwargs):
  4. # do import here to avoid loading inductor into memory when it is not used
  5. from torch._inductor.compile_fx import compile_fx
  6. return compile_fx(*args, **kwargs)