skyrim-vr-upscale / dev /include /registration.h
phanerozoic's picture
source: kernel, HLSL in-game path, OpenVR integration, benches
5506acb verified
Raw
History Blame
270 Bytes
// 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)