MCPcopy Create free account

hub / github.com/LaurentMazare/ug / functions

Functions526 in github.com/LaurentMazare/ug

⨍ Functions 526◇ Types & classes 143
↓ 4 callersMethodget
(&self, name: &str)
ug-core/src/safetensors.rs:281
↓ 4 callersMethodget_arg_id
(&self, arg_id: ArgId)
ug-core/src/schedule.rs:71
↓ 4 callersMethoditems
(&self)
ug-core/src/schedule.rs:107
↓ 4 callersMethodlower
(&self)
ug-pyo3/src/lib.rs:257
↓ 4 callersMethodmatmul
(&self, rhs: Self)
ug-core/src/lazy_buffer.rs:406
↓ 4 callersMethodmax
(lhs: Self, rhs: Self)
ug-core/src/interpreter.rs:54
↓ 4 callersMethodmul
(self, rhs: usize)
ug-core/src/lower_op.rs:78
↓ 4 callersMethodrun
(&self, f: &Self::Func, args: &mut [&mut Self::Slice])
ug-cuda/src/runtime.rs:266
↓ 4 callersMethodset
(&mut self, id: VarId, value: Value<N>)
ug-core/src/interpreter.rs:171
↓ 4 callersFunctionsoftmax
(src: &LB<D>)
ug-llama/src/model.rs:94
↓ 4 callersMethodsplit_dim
(&self, dim: I, size1: usize, size2: usize)
ug-core/src/lazy_buffer.rs:473
↓ 3 callersMethodadd
(&self, rhs: &Self)
ug-core/src/interpreter.rs:69
↓ 3 callersFunctionarg
(index: usize, dtype: DType)
ug-llama/src/model.rs:35
↓ 3 callersMethodas_const
(&self)
ug-core/src/lang.rs:231
↓ 3 callersMethodblock_axis
(&self)
ug-core/src/lower_op.rs:37
↓ 3 callersMethodcmp
(&self, other: &Self)
ug-core/src/const.rs:16
↓ 3 callersMethodcustom_ip
Applies a custom in-place operation.
ug-core/src/lazy_buffer.rs:313
↓ 3 callersMethoddefine_global
(index: usize, dtype: DType)
ug-pyo3/src/lib.rs:334
↓ 3 callersMethoddtype
(&self)
ug-cuda/src/runtime.rs:353
↓ 3 callersFunctionfmt_f32
(v: f32, f: &mut std::fmt::Formatter<'_>)
ug-metal/src/code_gen.rs:14
↓ 3 callersFunctionfmt_f32
(v: f32, f: &mut std::fmt::Formatter<'_>)
ug-core/src/cpu_code_gen.rs:14
↓ 3 callersFunctionfmt_f32
(v: f32, f: &mut std::fmt::Formatter<'_>)
ug-cuda/src/code_gen.rs:14
↓ 3 callersFunctionfunction
(obj, indent: str, text_signature: str = None)
stub.py:26
↓ 3 callersFunctiongemm_config
( alpha: T, beta: T, (b, m, n, k): (usize, usize, usize, usize), lhs_l: &Layout, rhs_l: &L
ug-cuda/src/gemm.rs:8
↓ 3 callersMethodget_id
(&self, id: VarId)
ug-core/src/interpreter.rs:180
↓ 3 callersMethodis_none
(&self)
ug-core/src/interpreter.rs:138
↓ 3 callersMethodkernel
(&self)
ug-core/src/schedule.rs:23
↓ 3 callersMethodlen
(&self)
ug-pyo3/src/lib.rs:64
↓ 3 callersMethodload_with_cast
( &self, name: &str, dtype: DType, device: &D, )
ug-core/src/safetensors.rs:73
↓ 3 callersMethodmerge_dims
Merge the dims dim and dim + 1 together.
ug-core/src/lazy_buffer.rs:496
↓ 3 callersMethodmin
(lhs: Self, rhs: Self)
ug-core/src/interpreter.rs:51
↓ 3 callersFunctionmm
(dev: &D)
ug-core/src/common_tests.rs:15
↓ 3 callersMethodpartial_cmp
(&self, other: &Self)
ug-core/src/const.rs:21
↓ 3 callersMethodptr
(dtype: DType)
ug-core/src/lang.rs:166
↓ 3 callersMethodpush_kernel
( &mut self, buffer: &LazyBuffer<D>, ast: Ast, dst_layout: Option<Layout>,
ug-core/src/schedule.rs:440
↓ 3 callersMethodrealized
(&self)
ug-core/src/lazy_buffer.rs:146
↓ 3 callersMethodreduce
(&self, op: crate::lang::ReduceOp, dim: I)
ug-core/src/lazy_buffer.rs:453
↓ 3 callersMethodreshape
(&self, s: S)
ug-core/src/lazy_buffer.rs:546
↓ 3 callersFunctionsimple_dotprod
(vec_len: usize)
ug-core/src/samples.rs:29
↓ 3 callersMethodsize_in_bytes
(&self)
ug-core/src/lang.rs:74
↓ 3 callersFunctionsoftmax
(dim1: usize, dim2: usize)
ug-core/src/samples.rs:230
↓ 3 callersMethodstride
(&self)
ug-core/src/lang.rs:264
↓ 3 callersMethodsynchronize
(&self)
ug-cuda/src/runtime.rs:206
↓ 3 callersMethodthread_block
(&self)
ug-core/src/lower_op.rs:41
↓ 3 callersFunctionv
(id: usize)
ug-pyo3/src/lib.rs:269
↓ 3 callersMethodwith_block_axis
(mut self, axis: usize)
ug-core/src/lower_op.rs:27
↓ 3 callersMethodwith_thread_block
(mut self, dim: usize, block_dim: usize)
ug-core/src/lower_op.rs:32
↓ 3 callersMethodwrite_str
(&mut self, s: &str)
ug-core/src/display.rs:146
↓ 2 callersMethodadd
(self, rhs: Self)
ug-core/src/lower_op.rs:73
↓ 2 callersMethodas_str
(&self)
ug-core/src/lang.rs:48
↓ 2 callersFunctionbroadcast_binary
(op: BinaryOp, lhs: Ast, rhs: Ast)
ug-core/src/lang.rs:522
↓ 2 callersMethodcan_be_compressed
(&self)
ug-core/src/layout.rs:585
↓ 2 callersMethodcompile_c
(&self, c_code: &[u8], func_name: String)
ug-core/src/cpu_runtime.rs:467
↓ 2 callersMethodcompile_with_cache
( &self, compilation_cache: &mut crate::cache::CompilationCache<D>, )
ug-core/src/schedule.rs:115
↓ 2 callersMethodcompress_all
(&self)
ug-core/src/layout.rs:599
↓ 2 callersMethodconst_i32
(v: i32)
ug-pyo3/src/lib.rs:288
↓ 2 callersMethoddata_vec
(&self)
ug-core/src/lazy_buffer.rs:170
↓ 2 callersMethoddiv
(&self, rhs: &Self)
ug-core/src/interpreter.rs:78
↓ 2 callersFunctiondo_black
(content, is_pyi)
stub.py:160
↓ 2 callersFunctiondo_indent
(text: Optional[str], indent: str)
stub.py:20
↓ 2 callersMethodelem_count
(&self)
ug-core/src/lazy_buffer.rs:197
↓ 2 callersFunctionexp
(block_size: usize)
ug-core/src/samples.rs:52
↓ 2 callersMethodexp
(&self)
ug-pyo3/src/lib.rs:179
↓ 2 callersFunctionexp_block
(dim2: usize, block_size: usize)
ug-core/src/samples.rs:73
↓ 2 callersMethodgemm
( &self, dst: &mut [T], lhs: &[T], lhs_l: &Layout, rhs: &[T],
ug-core/src/cpu_runtime.rs:551
↓ 2 callersFunctiongen
(w: &mut W, func_name: &str, kernel: &ssa::Kernel)
ug-metal/src/code_gen.rs:68
↓ 2 callersFunctiongen
(w: &mut W, func_name: &str, kernel: &ssa::Kernel)
ug-core/src/cpu_code_gen.rs:68
↓ 2 callersFunctionget_module_members
(module)
stub.py:75
↓ 2 callersMethodlaunch3
(&self, s1: &Slice, s2: &Slice, s3: &mut Slice)
ug-pyo3/src/lib.rs:29
↓ 2 callersMethodlaunch_config
(&self)
ug-core/src/lang.rs:719
↓ 2 callersMethodload_slice
(&self, name: &str, device: &D)
ug-core/src/safetensors.rs:83
↓ 2 callersMethodlower
(&self, idxs: &Indexes)
ug-core/src/lower_op.rs:206
↓ 2 callersMethodmatmul_
(&self, rhs: Self, transpose: bool)
ug-core/src/lazy_buffer.rs:414
↓ 2 callersMethodmatmul_t
(&self, rhs: Self)
ug-core/src/lazy_buffer.rs:410
↓ 2 callersMethodmod_
(&self, rhs: &Self)
ug-core/src/interpreter.rs:81
↓ 2 callersMethodmul
(&self, rhs: &Self)
ug-core/src/interpreter.rs:72
↓ 2 callersMethodneg
(&self)
ug-core/src/interpreter.rs:25
↓ 2 callersMethodnew_command_queue
(&self)
ug-metal/src/runtime.rs:58
↓ 2 callersMethodop
(&self)
ug-core/src/lazy_buffer.rs:126
↓ 2 callersMethodout_of_range
(&self, shape: &Shape, op: &'static str)
ug-core/src/layout.rs:277
↓ 2 callersFunctionrepeat
(lb: &LB<D>, axis: usize, n_rep: usize)
ug-llama/src/model.rs:80
↓ 2 callersMethodset_l
(&self, values: Self, dst_layout: crate::Layout)
ug-core/src/lazy_buffer.rs:352
↓ 2 callersFunctionsoftmax_block
(_dim1: usize, dim2: usize, block_size: usize)
ug-core/src/samples.rs:185
↓ 2 callersMethodstriding_error
(&self, lhs_l: &Layout, rhs_l: &Layout, msg: &'static str)
ug-core/src/cpu_runtime.rs:517
↓ 2 callersMethodsub
(&self, rhs: &Self)
ug-core/src/interpreter.rs:75
↓ 2 callersFunctionwrite
(module, directory, origin, check=False)
stub.py:173
↓ 1 callersMethodab_skip
(&self, lhs_l: &Layout, rhs_l: &Layout)
ug-core/src/cpu_runtime.rs:527
↓ 1 callersMethodallocate_uninit
(&self, dtype: ug::DType, len: usize)
ug-cuda/src/runtime.rs:217
↓ 1 callersMethodarg_ids
We use a BTreeSet rather than a HashSet here to ensure consistent ordering when converting the result to a Vec.
ug-core/src/lang.rs:550
↓ 1 callersMethodas_mut_ref
(&mut self)
ug-core/src/cpu_runtime.rs:118
↓ 1 callersMethodas_str
String representation for dtypes.
ug-core/src/dtype.rs:40
↓ 1 callersMethodas_usize
(&self)
ug-metal/src/runtime.rs:31
↓ 1 callersMethodas_usize
(&self)
ug-core/src/lang.rs:99
↓ 1 callersMethodas_usize
(&self)
ug-core/src/cpu_runtime.rs:426
↓ 1 callersMethodas_usize
(&self)
ug-cuda/src/runtime.rs:44
↓ 1 callersMethodbroadcast_shape_binary_op
Check whether the two shapes are compatible for broadcast, and if it is the case return the broadcasted shape. This is to be used for binary pointwise
ug-core/src/layout.rs:200
↓ 1 callersFunctioncall_mlx_gemm
( device: &Device, encoder: &metal::ComputeCommandEncoderRef, dtype: GemmDType, (b, m, n, k):
ug-metal/src/runtime.rs:385
↓ 1 callersMethodcan_merge_all
When the result at index i is true, the dims i and i + 1 can be merged.
ug-core/src/lower_op.rs:636
↓ 1 callersMethodcast
(&self, dtype: DType)
ug-core/src/lazy_buffer.rs:243
↓ 1 callersMethodcmp
(&self, other: &Self)
ug-core/src/lang.rs:151
← previousnext →101–200 of 526, ranked by callers