1#pragma once
2
3#include <c10/macros/Export.h>
4#include <kernel.h>
5
6#include <string>
7
8namespace torch {
9namespace jit {
10namespace fuser {
11namespace cuda {
12namespace codegen {
13
14//! Generates a CUDA kernel definition for the given kernel
15TORCH_CUDA_CU_API std::string generateCudaKernel(
16 const kir::Kernel* kernel,
17 const std::string& kernel_name = "CUDAGeneratedKernel");
18
19} // namespace codegen
20} // namespace cuda
21} // namespace fuser
22} // namespace jit
23} // namespace torch
24