#include <ckernel_kernel_specs.h>

Data Fields
const char *	alias_of

const char *	condition

CKDataType	dtype

const char *	name

int	optional

CKBufferRole	role

CKBufferScope	scope

CKDimToken	shape [4]

Detailed Description

Definition at line 45 of file ckernel_kernel_specs.h.

Field Documentation

◆ alias_of

const char* CKBufferSpec::alias_of

Definition at line 51 of file ckernel_kernel_specs.h.

Referenced by ck_ir_v2_build_decoder_backward(), ck_ir_v2_copy_buffer_spec(), emit_global_aliases_to_layer(), emit_global_allocations(), and emit_sgd_update().

◆ condition

const char* CKBufferSpec::condition

Definition at line 52 of file ckernel_kernel_specs.h.

Referenced by ck_ir_v2_build_decoder_backward(), ck_ir_v2_copy_buffer_spec(), emit_global_allocations(), and emit_layer_allocations().

◆ dtype

CKDataType CKBufferSpec::dtype

Definition at line 53 of file ckernel_kernel_specs.h.

Referenced by ck_ir_v2_build_decoder_backward(), and ck_ir_v2_copy_buffer_spec().

◆ name

const char* CKBufferSpec::name

Definition at line 46 of file ckernel_kernel_specs.h.

Referenced by ck_buffer_uses_weight_dtype(), ck_first_layer_buffer_name(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_copy_buffer_spec(), ck_weight_dtype_expr(), emit_global_aliases_to_layer(), emit_global_allocations(), emit_global_offset_fields(), emit_layer_allocations(), emit_layer_offsets_struct(), emit_sgd_update(), and emit_zero_grad().

◆ optional

int CKBufferSpec::optional

Definition at line 50 of file ckernel_kernel_specs.h.

Referenced by ck_ir_v2_build_decoder_backward(), and ck_ir_v2_copy_buffer_spec().

◆ role

CKBufferRole CKBufferSpec::role

Definition at line 48 of file ckernel_kernel_specs.h.

Referenced by ck_buffer_should_alloc(), ck_buffer_uses_weight_dtype(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_copy_buffer_spec(), ck_weight_dtype_expr(), emit_global_allocations(), emit_layer_allocations(), emit_sgd_update(), and emit_zero_grad().

◆ scope

CKBufferScope CKBufferSpec::scope

Definition at line 47 of file ckernel_kernel_specs.h.

Referenced by ck_first_layer_buffer_name(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_copy_buffer_spec(), ck_weight_dtype_expr(), emit_global_aliases_to_layer(), emit_global_allocations(), emit_global_offset_fields(), emit_layer_allocations(), emit_layer_offsets_struct(), emit_sgd_update(), and emit_zero_grad().

◆ shape

CKDimToken CKBufferSpec::shape[4]

Definition at line 49 of file ckernel_kernel_specs.h.

Referenced by ck_ir_v2_build_decoder_backward(), ck_ir_v2_copy_buffer_spec(), ck_ir_v2_parse_buffers(), emit_global_allocations(), emit_layer_allocations(), emit_sgd_update(), and emit_zero_grad().

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

ckernel_kernel_specs.h

Data Fields