Halide
Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU > Class Template Reference

A code generator that emits GPU code from a given Halide stmt. More...

#include <CodeGen_GPU_Host.h>

Inherits CodeGen_CPU.

Public Member Functions

 CodeGen_GPU_Host (Target)
 Create a GPU code generator. More...
 
 ~CodeGen_GPU_Host () override
 

Protected Member Functions

void compile_func (const LoweredFunc &func, const std::string &simple_name, const std::string &extern_name) override
 
void visit (const For *) override
 Nodes for which we need to override default behavior for the GPU runtime. More...
 
llvm::Value * get_module_state (const std::string &api_unique_name, bool create=true)
 

Protected Attributes

std::string function_name
 

Detailed Description

template<typename CodeGen_CPU>
class Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU >

A code generator that emits GPU code from a given Halide stmt.

Definition at line 27 of file CodeGen_GPU_Host.h.

Constructor & Destructor Documentation

◆ CodeGen_GPU_Host()

template<typename CodeGen_CPU >
Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU >::CodeGen_GPU_Host ( Target  )

Create a GPU code generator.

GPU target is selected via CodeGen_GPU_Options. Processor features can be enabled using the appropriate flags from Target

◆ ~CodeGen_GPU_Host()

template<typename CodeGen_CPU >
Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU >::~CodeGen_GPU_Host ( )
override

Member Function Documentation

◆ compile_func()

template<typename CodeGen_CPU >
void Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU >::compile_func ( const LoweredFunc func,
const std::string &  simple_name,
const std::string &  extern_name 
)
overrideprotected

◆ visit()

template<typename CodeGen_CPU >
void Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU >::visit ( const For )
overrideprotected

Nodes for which we need to override default behavior for the GPU runtime.

◆ get_module_state()

template<typename CodeGen_CPU >
llvm::Value* Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU >::get_module_state ( const std::string &  api_unique_name,
bool  create = true 
)
protected

Member Data Documentation

◆ function_name

template<typename CodeGen_CPU >
std::string Halide::Internal::CodeGen_GPU_Host< CodeGen_CPU >::function_name
protected

Definition at line 72 of file CodeGen_GPU_Host.h.


The documentation for this class was generated from the following file: