Functions — github.com/LaurentMazare/ug

hub / github.com/LaurentMazare/ug / functions

Functions526 in github.com/LaurentMazare/ug

⨍ Functions 526◇ Types & classes 143

↓ 180 callersMethodclone

(&self)

ug-core/src/lazy_buffer.rs:22

↓ 132 callersMethodpush

(&mut self, inst: SsaI)

ug-core/src/block.rs:108

↓ 61 callersMethodlen

(&self)

ug-core/src/lang.rs:261

↓ 57 callersMethodto_a

(self)

ug-core/src/block.rs:22

↓ 54 callersMethoddims

The dimensions as a slice of `usize`.

ug-core/src/layout.rs:135

↓ 45 callersMethodbinary

(&self, rhs: &Self, op: BinaryOp)

ug-core/src/lang.rs:323

↓ 40 callersMethoddata

(&self)

ug-core/src/lazy_buffer.rs:166

↓ 40 callersMethoddevice

(&self)

ug-core/src/lazy_buffer.rs:185

↓ 36 callersMethodw

(self)

ug-cuda/src/runtime.rs:14

↓ 35 callersMethodcopy_host_to_device

(&mut self, src: &[DT])

ug-core/src/cpu_runtime.rs:376

↓ 35 callersMethoddtype

(&self)

ug-core/src/lang.rs:285

↓ 33 callersFunctioncst

(v: I)

ug-core/src/lang.rs:529

↓ 31 callersMethodget

(&self, kernel: &NormalizedKernel)

ug-core/src/cache.rs:86

↓ 31 callersMethodshape

(&self)

ug-core/src/lang.rs:544

↓ 31 callersMethodto_index

(&self, shape: &Shape, op: &'static str)

ug-core/src/layout.rs:232

↓ 30 callersMethoddevice_ptr_mut

( &'a mut self, stream: &'a Arc<CudaStream>, )

ug-cuda/src/runtime.rs:149

↓ 28 callersMethodrank

The rank is the number of dimensions, 0 for a scalar value, 1 for a vector, etc.

ug-core/src/layout.rs:126

↓ 27 callersFunctionwalk

(ast: &Ast, tgt_dim: usize, accs: &mut Vec<(Id, Ast)>)

ug-core/src/lower_op.rs:255

↓ 26 callersMethodinsert

(&mut self, kernel: NormalizedKernel, func: std::sync::Arc<D::Func>)

ug-core/src/cache.rs:90

↓ 26 callersMethodshape

(&self)

ug-core/src/lazy_buffer.rs:193

↓ 24 callersMethodto_varid

(self)

ug-core/src/block.rs:18

↓ 23 callersMethodto_vec

(&self)

ug-core/src/lib.rs:42

↓ 22 callersMethodas_ref

(&self)

ug-core/src/cpu_runtime.rs:108

↓ 22 callersMethodbuffer

(&self)

ug-metal/src/runtime.rs:221

↓ 22 callersMethodid

(&self)

ug-core/src/lang.rs:171

↓ 20 callersMethodas_mut_ptr

(&mut self)

ug-core/src/cpu_runtime.rs:64

↓ 16 callersMethoddims

(&self)

ug-core/src/lazy_buffer.rs:201

↓ 15 callersFunctionload

(src: ArgId, layout: Layout, dtype: DType)

ug-core/src/lang.rs:446

↓ 15 callersMethodoffset

(&self)

ug-core/src/lang.rs:258

↓ 14 callersFunctionhole_size

(el_count: usize, prod_d: usize, s: &dyn std::fmt::Debug)

ug-core/src/layout.rs:442

↓ 14 callersMethodlen

(&self)

ug-metal/src/runtime.rs:196

↓ 14 callersMethodlower

( &self, range_id: Id, per_arg: &std::collections::HashMap<lang::ArgId, ssa::VarId>,

ug-core/src/lower.rs:7

↓ 14 callersMethodstrides

(&self)

ug-core/src/layout.rs:652

↓ 13 callersMethodget

(&self, a: ssa::A)

ug-core/src/interpreter.rs:188

↓ 13 callersMethodnum_elements

(&self)

ug-core/src/layout.rs:117

↓ 12 callersMethodunary

(&self, op: UnaryOp)

ug-core/src/lang.rs:319

↓ 11 callersFunctionbinary

(op: BinaryOp, lhs: Ast, rhs: Ast)

ug-core/src/lang.rs:507

↓ 10 callersFunctionarg

(index: usize, dtype: DType)

ug-core/src/samples.rs:6

↓ 10 callersMethodbroadcast

(&self, s: S)

ug-core/src/lazy_buffer.rs:591

↓ 10 callersMethodbt

(self)

ug-core/src/error.rs:69

↓ 10 callersMethodbuilder

(&self)

ug-cuda/src/runtime.rs:61

↓ 10 callersMethodcompile_cu

(&self, cu_code: &str, func_name: &str)

ug-cuda/src/runtime.rs:179

↓ 10 callersMethodslice

(&self)

ug-cuda/src/runtime.rs:130

↓ 9 callersMethodas_usize

(&self)

ug-core/src/interpreter.rs:15

↓ 9 callersMethoddevice_ptr

( &'a self, stream: &'a Arc<CudaStream>, )

ug-cuda/src/runtime.rs:136

↓ 9 callersMethodlen

(&self)

ug-core/src/cpu_runtime.rs:84

↓ 9 callersMethodpush_schedule_item

(&mut self, buffer: &LazyBuffer<D>)

ug-core/src/schedule.rs:471

↓ 9 callersFunctionrun

(dev: &D, args: &Args)

ug-llama/src/main.rs:104

↓ 9 callersMethodto_vec

(&self)

ug-metal/src/runtime.rs:204

↓ 8 callersFunctiona

(id: usize)

ug-pyo3/src/lib.rs:273

↓ 8 callersMethodas_ptr

(&mut self)

ug-core/src/cpu_runtime.rs:74

↓ 8 callersFunctioncat

(dev: &D)

ug-core/src/common_tests.rs:24

↓ 8 callersMethodlen

(&self)

ug-cuda/src/runtime.rs:395

↓ 8 callersMethodload

( ptr: &Arg, offset: &IndexExprNode, len: &IndexExprNode, stride: &IndexExprNo

ug-core/src/lang.rs:300

↓ 8 callersFunctionreduce

(op: ReduceOp, arg: Ast, dim: usize)

ug-core/src/lang.rs:496

↓ 8 callersMethodrelocate

This switches all the VarId to be in line number rather than "random" unique identifiers.

ug-core/src/block.rs:143

↓ 8 callersMethodtype_

(&self)

ug-core/src/lang.rs:175

↓ 7 callersMethodargs

(&self)

ug-core/src/lang.rs:727

↓ 7 callersMethodas_f32

(&self)

ug-core/src/interpreter.rs:122

↓ 7 callersMethodcompile

(&self)

ug-core/src/schedule.rs:111

↓ 7 callersMethoddims4

(&self)

ug-core/src/lazy_buffer.rs:221

↓ 7 callersFunctiongen

(w: &mut W, func_name: &str, kernel: &ssa::Kernel)

ug-cuda/src/code_gen.rs:68

↓ 7 callersMethodinstrs

(&self)

ug-core/src/lang.rs:723

↓ 7 callersMethodmul

(&self, rhs: &Self)

ug-core/src/lang.rs:227

↓ 7 callersMethodnew_command_buffer

(&self)

ug-metal/src/runtime.rs:62

↓ 7 callersMethodrange

(&mut self, lo: L, up: U, step: usize)

ug-core/src/block.rs:54

↓ 7 callersMethodrank

(&self)

ug-core/src/lazy_buffer.rs:225

↓ 7 callersMethodrun

(&self)

ug-core/src/schedule.rs:234

↓ 7 callersFunctionsimple_add

(vec_len: usize)

ug-core/src/samples.rs:10

↓ 7 callersMethodslice_from_values

(&self, vs: &[D])

ug-cuda/src/runtime.rs:200

↓ 7 callersFunctionstore

(dst: ArgId, layout: Layout, value: Ast)

ug-core/src/lang.rs:579

↓ 7 callersMethodtranspose

(&self)

ug-core/src/layout.rs:624

↓ 7 callersFunctionunary

(op: UnaryOp, arg: Ast)

ug-core/src/lang.rs:486

↓ 6 callersMethodadd

(&self, rhs: &Self)

ug-core/src/lang.rs:223

↓ 6 callersMethodas_i32

(&self)

ug-core/src/interpreter.rs:129

↓ 6 callersFunctionbroadcast

(arg: Ast, shape: S)

ug-core/src/lang.rs:452

↓ 6 callersMethodcompile_metal

( &self, metal_code: &str, func_name: &str, launch_config: ug::lang::LaunchCon

ug-metal/src/runtime.rs:75

↓ 6 callersMethodcst

(v: usize)

ug-core/src/lang.rs:215

↓ 6 callersMethoddtype

(&self)

ug-core/src/lazy_buffer.rs:189

↓ 6 callersMethodpipeline

(&self)

ug-metal/src/runtime.rs:44

↓ 6 callersMethodrealize

(&self)

ug-core/src/lazy_buffer.rs:156

↓ 6 callersMethodwith_path

(self, p: P)

ug-core/src/error.rs:78

↓ 6 callersMethodzeros

(&self, len: usize)

ug-cuda/src/runtime.rs:195

↓ 5 callersFunctionadd

(dev: &D)

ug-core/src/common_tests.rs:6

↓ 5 callersMethodallocate_uninit

(&self, dtype: DType, len: usize)

ug-core/src/cpu_runtime.rs:255

↓ 5 callersMethodcudarc_stream

(&self)

ug-cuda/src/runtime.rs:171

↓ 5 callersMethoddtype

(&self)

ug-core/src/cpu_runtime.rs:98

↓ 5 callersMethodend_range

(&mut self, range: Range)

ug-core/src/block.rs:62

↓ 5 callersMethodfmt

(&self, f: &mut std::fmt::Formatter)

ug-core/src/display.rs:44

↓ 5 callersMethodfmt_tensor

( &self, t: &LB<D>, indent: usize, max_w: usize, summarize: bool,

ug-core/src/display.rs:177

↓ 5 callersMethodhead_dim

(&self)

ug-llama/src/model.rs:27

↓ 5 callersMethodis_empty

(&self)

ug-core/src/cpu_runtime.rs:94

↓ 5 callersMethodlaunch_cfg

(&self)

ug-cuda/src/runtime.rs:65

↓ 5 callersMethodmax_width

(&self, to_display: &LB<D>)

ug-core/src/display.rs:157

↓ 5 callersMethodmaybe_allocate_uninit

(&self)

ug-core/src/lazy_buffer.rs:116

↓ 5 callersMethodnarrow

(&self, dim: D1, offset: usize, length: usize)

ug-core/src/lazy_buffer.rs:568

↓ 5 callersMethodptr_f32

()

ug-pyo3/src/lib.rs:119

↓ 5 callersMethodstore

( dst_ptr: &Arg, dst_offset: &IndexExprNode, dst_len: &IndexExprNode, dst_stri

ug-core/src/lang.rs:354

↓ 5 callersMethodw

(self)

ug-metal/src/runtime.rs:15

↓ 4 callersMethoddims2

(&self)

ug-core/src/lazy_buffer.rs:213

next →1–100 of 526, ranked by callers