TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) { | |
// Activation ops | |
ops.def("dummy() -> ()"); | |
} | |
REGISTER_EXTENSION(TORCH_EXTENSION_NAME) | |
TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) { | |
// Activation ops | |
ops.def("dummy() -> ()"); | |
} | |
REGISTER_EXTENSION(TORCH_EXTENSION_NAME) | |