CKIRV2Graph Struct Reference

#include <ckernel_ir_v2.h>

Data Fields
CKIRV2Buffer *	buffers
CKModelConfig	config
int	fused_qkv
int	gated_mlp
int	has_pos_emb
CKIRV2Node *	nodes
int	num_buffers
int	num_nodes
int	tie_word_embeddings

Detailed Description

Definition at line 55 of file ckernel_ir_v2.h.

Field Documentation

buffers

CKIRV2Buffer* CKIRV2Graph::buffers

Definition at line 62 of file ckernel_ir_v2.h.

Referenced by build_plan(), ck_codegen_v2_emit_struct(), ck_ir_v2_apply_meta(), ck_ir_v2_apply_weight_dtypes(), ck_ir_v2_build_decoder(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_emit_memory_plan(), ck_ir_v2_find_buffer_index(), ck_ir_v2_free(), ck_ir_v2_lower_copy_buffers(), ck_ir_v2_parse_buffers(), ck_ir_v2_serialize_json_internal(), emit_body_fields(), emit_body_values(), emit_footer_fields(), emit_footer_values(), emit_header_fields(), emit_header_values(), emit_schedule_block(), and find_buffer_by_name().

config

CKModelConfig CKIRV2Graph::config

Definition at line 56 of file ckernel_ir_v2.h.

Referenced by buffer_enabled(), build_plan(), ck_codegen_v2_emit_sections(), ck_ir_v2_build_decoder(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_lower_emit_json(), ck_ir_v2_lower_graph(), ck_ir_v2_parse_json(), ck_ir_v2_serialize_json_internal(), and main().

fused_qkv

int CKIRV2Graph::fused_qkv

Definition at line 59 of file ckernel_ir_v2.h.

Referenced by ck_ir_v2_apply_meta(), ck_ir_v2_build_decoder(), ck_ir_v2_lower_graph(), ck_ir_v2_parse_json(), and ck_ir_v2_serialize_json_internal().

gated_mlp

int CKIRV2Graph::gated_mlp

Definition at line 60 of file ckernel_ir_v2.h.

Referenced by ck_ir_v2_apply_meta(), ck_ir_v2_build_decoder(), ck_ir_v2_lower_graph(), ck_ir_v2_parse_json(), and ck_ir_v2_serialize_json_internal().

has_pos_emb

int CKIRV2Graph::has_pos_emb

Definition at line 57 of file ckernel_ir_v2.h.

Referenced by buffer_enabled(), ck_ir_v2_apply_meta(), ck_ir_v2_build_decoder(), ck_ir_v2_lower_graph(), ck_ir_v2_parse_json(), and ck_ir_v2_serialize_json_internal().

nodes

CKIRV2Node* CKIRV2Graph::nodes

Definition at line 64 of file ckernel_ir_v2.h.

Referenced by ck_ir_v2_build_decoder(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_free(), ck_ir_v2_lower_copy_nodes(), ck_ir_v2_parse_nodes(), ck_ir_v2_serialize_json_internal(), and emit_schedule_block().

num_buffers

int CKIRV2Graph::num_buffers

Definition at line 61 of file ckernel_ir_v2.h.

Referenced by build_plan(), ck_codegen_v2_emit_struct(), ck_ir_v2_build_decoder(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_emit_memory_plan(), ck_ir_v2_find_buffer_index(), ck_ir_v2_free(), ck_ir_v2_lower_copy_buffers(), ck_ir_v2_parse_buffers(), ck_ir_v2_serialize_json_internal(), emit_body_fields(), emit_body_values(), emit_footer_fields(), emit_footer_values(), emit_header_fields(), emit_header_values(), emit_schedule_block(), find_buffer_by_name(), and main().

num_nodes

int CKIRV2Graph::num_nodes

Definition at line 63 of file ckernel_ir_v2.h.

Referenced by ck_ir_v2_build_decoder(), ck_ir_v2_build_decoder_backward(), ck_ir_v2_free(), ck_ir_v2_lower_copy_nodes(), ck_ir_v2_parse_nodes(), ck_ir_v2_serialize_json_internal(), emit_schedule_block(), and main().

tie_word_embeddings

int CKIRV2Graph::tie_word_embeddings

Definition at line 58 of file ckernel_ir_v2.h.

Referenced by ck_ir_v2_apply_meta(), ck_ir_v2_build_decoder(), ck_ir_v2_lower_graph(), ck_ir_v2_parse_json(), and ck_ir_v2_serialize_json_internal().

---

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

• ckernel_ir_v2.h