File size: 339 Bytes
79bc4c6 | 1 2 3 4 5 6 7 8 | // Shim for kernel-builder's registration.h, for local JIT builds.
// cpp_extension.load(is_python_module=False) loads via torch.ops.load_library,
// so TORCH_LIBRARY runs at dlopen and no PyInit module object is needed.
#pragma once
#define TORCH_LIBRARY_EXPAND(NAME, MODULE) TORCH_LIBRARY(NAME, MODULE)
#define REGISTER_EXTENSION(NAME)
|