File size: 270 Bytes
5506acb | 1 2 3 4 5 6 7 | // Shim for kernel-builder's registration.h in local JIT builds: the library
// loads via torch.ops.load_library, so no PyInit module object is needed.
#pragma once
#define TORCH_LIBRARY_EXPAND(NAME, MODULE) TORCH_LIBRARY(NAME, MODULE)
#define REGISTER_EXTENSION(NAME)
|