#pragma once #include "../jit/compiler.hpp" #include "../jit/device_runtime.hpp" namespace deep_gemm::runtime { // The init and other functions are now exposed via TORCH_LIBRARY in python_api.cpp } // namespace deep_gemm::runtime