torch-ext/registration.h

#pragma once

// Provide fallbacks so the extension builds even if the helper macros are absent.
#ifndef REGISTER_EXTENSION
#define REGISTER_EXTENSION(name)
#endif

#ifndef TORCH_LIBRARY_EXPAND
#define TORCH_LIBRARY_EXPAND(ns, m) TORCH_LIBRARY(ns, m)
#endif