shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / torch-ext /registration.h

leonardlin's picture

Add ROCm build artifacts and HIP backend

1e407f0 2 months ago

history blame contribute delete

259 Bytes

	#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