| """Load resample-poly natively on hosts without a published variant. |
| |
| import load_local |
| rp = load_local.load() |
| """ |
| import os |
| import sys |
| import types |
|
|
| _cached = None |
|
|
|
|
| def load(verbose=False): |
| """Return the resample_poly module, JIT-built from the local source.""" |
| global _cached |
| if _cached is not None: |
| return _cached |
| from torch.utils.cpp_extension import load as _jit |
|
|
| root = os.path.dirname(os.path.abspath(__file__)) |
| cflags = ["/O2"] if os.name == "nt" else ["-O3"] |
| ext = _jit(name="resample_poly_ext", |
| sources=[os.path.join(root, "local_bind.cpp"), |
| os.path.join(root, "resample_csrc", "resample.cpp")], |
| extra_include_paths=[os.path.join(root, "torch-ext")], |
| extra_cflags=cflags, |
| verbose=verbose) |
| ops_mod = types.ModuleType("resample_poly._ops") |
| ops_mod.ops = type("_Ops", (), { |
| "rp_resample": staticmethod(ext.rp_resample), |
| })() |
| sys.modules["resample_poly._ops"] = ops_mod |
| sys.path.insert(0, os.path.join(root, "torch-ext")) |
| import resample_poly |
| _cached = resample_poly |
| return resample_poly |
|
|
