RuntimeCompiler

#include <rtcompiler.hh>

class RuntimeCompiler

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:93

List of all members

Name	Kind	Owner
`get_dynamic_vm_kernel`	`function`	Declared here
`get_expr_kernel`	`function`	Declared here
`get_stencil_expr_kernel`	`function`	Declared here
`get_fill_kernel`	`function`	Declared here
`get_copy_kernel`	`function`	Declared here
`get_copy_view_into_view_kernel`	`function`	Declared here
`instance`	`function`	Declared here
`m_kernel_cache`	`variable`	Declared here
`RuntimeCompiler`	`function`	Declared here
`get_name_for_type`	`function`	Declared here
`to_sass`	`function`	Declared here
`get_dynamic_vm_kernel_str`	`function`	Declared here
`get_expression_kernel_str`	`function`	Declared here
`get_stencil_expr_kernel_str`	`function`	Declared here
`get_fill_kernel_str`	`function`	Declared here
`get_copy_kernel_str`	`function`	Declared here
`get_copy_view_into_view_kernel_str`	`function`	Declared here
`compile_kernel`	`function`	Declared here

Public Methods

Return	Name	Description
`CUfunction`	`get_dynamic_vm_kernel`
`CUfunction`	`get_expr_kernel`
`CUfunction`	`get_stencil_expr_kernel` `inline`
`CUfunction`	`get_fill_kernel`
`CUfunction`	`get_copy_kernel`
`CUfunction`	`get_copy_view_into_view_kernel`

get_dynamic_vm_kernel

CUfunction get_dynamic_vm_kernel(DType dest_t, bool expr_is_soarr)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:97

get_expr_kernel

CUfunction get_expr_kernel(DType dest_t, DType src_t, DType work_t, int n_views, int n_scalars, ssize_t ndim, const ssize_t * final_shape, const std::vector< Instruction > & instrs, bool expr_is_soarr = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:99

get_stencil_expr_kernel

inline

template<int NDIM> inline CUfunction get_stencil_expr_kernel(DType dest_t, DType src_t, DType work_t, const std::vector< StaticCoords< NDIM > > & offsets, const std::vector< Instruction > & instrs, const std::vector< Scalar > & scalars, const std::vector< std::uint8_t > & is_pointer_axis, bool expr_is_soarr = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:110

get_fill_kernel

CUfunction get_fill_kernel(DType dest_t, bool dest_is_so = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:153

get_copy_kernel

CUfunction get_copy_kernel(DType dest_t, DType src_t, bool src_is_so = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:154

get_copy_view_into_view_kernel

CUfunction get_copy_view_into_view_kernel(DType dest_t, DType src_t, bool dest_is_so = false, bool src_is_so = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:155

Public Static Methods

Return	Name	Description
`RuntimeCompiler &`	`instance` `static`

instance

static

static RuntimeCompiler & instance()

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:95

Private Attributes

Return	Name	Description
`std::unordered_map< std::string, CUfunction >`	`m_kernel_cache`

m_kernel_cache

std::unordered_map< std::string, CUfunction > m_kernel_cache

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:388

Private Methods

Return	Name	Description
	`RuntimeCompiler`
`std::string`	`get_name_for_type` `inline`
`CUfunction`	`to_sass`
`std::string`	`get_dynamic_vm_kernel_str`
`std::string`	`get_expression_kernel_str`
`std::string`	`get_stencil_expr_kernel_str` `inline`
`std::string`	`get_fill_kernel_str`
`std::string`	`get_copy_kernel_str`
`std::string`	`get_copy_view_into_view_kernel_str`
`std::string`	`compile_kernel`

RuntimeCompiler

RuntimeCompiler()

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:161

get_name_for_type

inline

template<typename DestT> inline std::string get_name_for_type()

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:164

to_sass

CUfunction to_sass(std::string cubin, const char * func_name)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:170

get_dynamic_vm_kernel_str

std::string get_dynamic_vm_kernel_str(DType dest_t, bool expr_is_soarr)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:172

get_expression_kernel_str

std::string get_expression_kernel_str(DType dest_t, DType src_t, DType work_t, int n_views, int n_scalars, ssize_t ndim, const ssize_t * final_shape, const std::vector< Instruction > & instrs, bool expr_is_soarr = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:174

get_stencil_expr_kernel_str

inline

template<int NDIM> inline std::string get_stencil_expr_kernel_str(DType dest_t, DType src_t, DType work_t, const std::vector< StaticCoords< NDIM > > & offsets, const std::vector< Instruction > & instrs, const std::vector< Scalar > & scalars, const std::vector< std::uint8_t > & is_pointer_axis, bool expr_is_soarr = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:185

get_fill_kernel_str

std::string get_fill_kernel_str(DType dest_t, bool dest_is_so = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:375

get_copy_kernel_str

std::string get_copy_kernel_str(DType dest_t, DType src_t, bool src_is_so = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:376

get_copy_view_into_view_kernel_str

std::string get_copy_view_into_view_kernel_str(DType dest_t, DType src_t, bool dest_is_so = false, bool src_is_so = false)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:377

compile_kernel

std::string compile_kernel(std::string kernel_str, std::string k_id, const char * func_name)

Defined in src/lib/ncarray/jit/device/rtcompiler.hh:382