MCPcopy Create free account

hub / github.com/LaurentMazare/ug / functions

Functions526 in github.com/LaurentMazare/ug

⨍ Functions 526◇ Types & classes 143
Methodbroadcast
(&self, shape: Vec<usize>)
ug-pyo3/src/lib.rs:174
Methodc_contiguous
(&self)
ug-core/src/layout.rs:664
Methodcat
(args: &[&Self], dim: D1)
ug-core/src/lazy_buffer.rs:722
Methodcat
(lhs: &LB<Self>, rhs: &LB<Self>, axis: usize)
ug-llama/src/cpu_ops.rs:128
Methodcat
(lhs: &LB<Self>, rhs: &LB<Self>, axis: usize)
ug-llama/src/metal_ops.rs:96
Methodcat
(lhs: &LB<Self>, rhs: &LB<Self>, axis: usize)
ug-llama/src/cuda_ops.rs:98
Methodcausal_mask
(src: &LB<Self>)
ug-llama/src/cpu_ops.rs:33
Methodcausal_mask
(_: &LB<Self>)
ug-llama/src/metal_ops.rs:179
Methodcausal_mask
(_: &LB<Self>)
ug-llama/src/cuda_ops.rs:176
Methodcompile
(&self, kernel: &ug::lang::ssa::Kernel, name: Option<&str>)
ug-metal/src/runtime.rs:158
Methodcompile
(&self, kernel: &crate::lang::ssa::Kernel, name: Option<&str>)
ug-core/src/cpu_runtime.rs:274
Methodcompile
(&self, kernel: &ug::lang::ssa::Kernel, name: Option<&str>)
ug-cuda/src/runtime.rs:247
Methodcompile_cu
( &self, cu_code: &str, func_name: &str, block_dim: u32, grid_dim: u32
ug-pyo3/src/lib.rs:438
Methodcompile_inner
( c_code: &[u8], func_name: String, tmp_c: &PathBuf, tmp_so: &
ug-core/src/cpu_runtime.rs:468
Methodcompile_ptx
( &self, ptx_code: &str, func_name: &str, block_dim: u32, grid_dim: u3
ug-pyo3/src/lib.rs:418
Methodconst_default
We cannot use the default trait as it's not const.
ug-core/src/display.rs:74
Methodconst_f32
(v: f32)
ug-pyo3/src/lib.rs:293
Functionconvert_slice
(data: &[u8])
ug-core/src/safetensors.rs:295
Methodcopy
( data: R, s: S, device: &D, )
ug-core/src/lazy_buffer.rs:687
Methodcopy_device_to_host
(&self, dst: &mut [DT])
ug-metal/src/runtime.rs:276
Methodcopy_device_to_host
(&self, dst: &mut [DT])
ug-cuda/src/runtime.rs:381
Methodcopy_host_to_device
(&mut self, src: &[DT])
ug-metal/src/runtime.rs:263
Methodcreate
(buffers: &[&LazyBuffer<D>])
ug-core/src/schedule.rs:78
Methodcreate_one
(buffer: &LazyBuffer<D>)
ug-core/src/schedule.rs:103
Methodcst
( c: C, s: S, device: &D, )
ug-core/src/lazy_buffer.rs:643
Methodcst
(&mut self, c: C)
ug-core/src/block.rs:104
Methodcst
(v: usize)
ug-pyo3/src/lib.rs:538
Methodcustom
( f: F, args: Vec<Self>, s: S, dtype: DType, device: &D, )
ug-core/src/lazy_buffer.rs:292
Methodcustom_softmax
(src: &LB<Self>)
ug-llama/src/cpu_ops.rs:6
Methodcustom_softmax
(src: &LB<Self>)
ug-llama/src/metal_ops.rs:146
Methodcustom_softmax
(src: &LB<Self>)
ug-llama/src/cuda_ops.rs:149
Methoddata
(&self)
ug-core/src/cpu_runtime.rs:128
Methoddata_mut
(&mut self)
ug-core/src/cpu_runtime.rs:132
Methoddebug
(err: impl std::fmt::Debug)
ug-core/src/error.rs:65
Methoddefault
()
ug-core/src/cache.rs:80
Methoddefine_acc_f32
(v: f32)
ug-pyo3/src/lib.rs:303
Methoddefine_acc_i32
(v: i32)
ug-pyo3/src/lib.rs:298
Methodderef
(&self)
ug-core/src/lazy_buffer.rs:16
Methodderef
(&self)
ug-core/src/const.rs:32
Methoddevice
(&self)
ug-metal/src/runtime.rs:246
Methoddevice
(&self)
ug-core/src/cpu_runtime.rs:372
Methoddevice
(&self)
ug-cuda/src/runtime.rs:363
Methoddims0
(&self)
ug-core/src/lazy_buffer.rs:205
Methoddims1
(&self)
ug-core/src/lazy_buffer.rs:209
Functiondisplay_multi_dim
()
ug-core/tests/display_tests.rs:52
Functiondisplay_scalar
()
ug-core/tests/display_tests.rs:10
Functiondisplay_vector
()
ug-core/tests/display_tests.rs:28
Methoddtype
(&self)
ug-metal/src/runtime.rs:242
Methoddtype
(&self)
ug-core/src/const.rs:135
Methodempty
()
ug-core/src/block.rs:134
Methodend_range
(start_idx: usize)
ug-pyo3/src/lib.rs:308
Functioneval_add
()
ug-core/tests/basic_tests.rs:4
Functioneval_dotprod
()
ug-core/tests/basic_tests.rs:16
Functioneval_ssa
( kernel: &Kernel, buffers: Vec<&mut Buffer>, _args: &[Value<N>], grid_idx: usize, )
ug-core/src/interpreter.rs:201
Methodextend
Modifies the shape by adding a list of additional dimensions at the end of the existing dimensions.
ug-core/src/layout.rs:193
Functionextract_additional_types
(module)
stub.py:215
Methodf16
()
ug-pyo3/src/lib.rs:92
Functionf32
(v: f32)
ug-pyo3/src/lib.rs:222
Methodf32
()
ug-pyo3/src/lib.rs:76
Methodflops_and_mem
(&self)
ug-pyo3/src/lib.rs:387
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
ug-metal/src/code_gen.rs:7
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
ug-core/src/lazy_buffer.rs:9
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
ug-core/src/cpu_code_gen.rs:7
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
ug-core/src/dtype.rs:17
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
ug-core/src/lang.rs:816
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
ug-core/src/layout.rs:10
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
ug-cuda/src/code_gen.rs:7
Methodfmt_dt
( &self, f: &mut std::fmt::Formatter, )
ug-core/src/display.rs:8
Functionfn_predicate
(obj)
stub.py:66
Methodfor_num_elems
(n: u32)
ug-core/src/lang.rs:42
Methodfrom
(value: i64)
ug-core/src/const.rs:113
Methodfrom
(value: Id)
ug-core/src/lower_op.rs:61
Methodfrom
(value: DType)
ug-core/src/lang.rs:68
Methodfrom
(dims: &[usize; C])
ug-core/src/layout.rs:16
Methodfrom
(value: Vec<bf16>)
ug-core/src/cpu_runtime.rs:34
Methodfrom
(_value: std::sync::PoisonError<T>)
ug-core/src/error.rs:84
Methodfrom
(val: Id)
ug-core/src/block.rs:32
Methodfrom_dims
(dims: &[usize])
ug-core/src/layout.rs:121
Methodfrom_expr
(expr: IndexExpr)
ug-core/src/lang.rs:209
Methodfrom_instrs
(instrs: Vec<Instr>)
ug-core/src/lang.rs:739
Methodfrom_shape
(shape: S)
ug-core/src/layout.rs:612
Methodfrom_slice
(data: D::Slice, s: S)
ug-core/src/lazy_buffer.rs:665
Methodfrom_str
(s: &str)
ug-core/src/dtype.rs:26
Methodfrom_usize
(v: usize)
ug-core/src/lang.rs:95
Methodfrom_varid
(v: ssa::VarId)
ug-core/src/block.rs:26
Methodget
(&self, index: usize)
ug-core/src/lazy_buffer.rs:535
Methodhash
(&self, state: &mut H)
ug-metal/src/runtime.rs:312
Methodhash
(&self, state: &mut H)
ug-core/src/const.rs:26
Functioni32
(v: i32)
ug-pyo3/src/lib.rs:217
Methodi32
()
ug-pyo3/src/lib.rs:84
Methodi64
()
ug-pyo3/src/lib.rs:80
Methodid
(&self)
ug-core/src/lazy_buffer.rs:469
Methodid
(&self)
ug-core/src/block.rs:48
Methodinto_ast
(self)
ug-core/src/schedule.rs:15
Methodinto_dims
(self)
ug-core/src/layout.rs:130
Methodinto_shape
(self, _el_count: usize)
ug-core/src/layout.rs:431
Methodis_contiguous
Returns true if the strides are C contiguous (aka row major).
ug-core/src/layout.rs:162
Methodis_empty
(&self)
ug-metal/src/runtime.rs:200
Methodis_empty
(&self)
ug-core/src/lib.rs:38
Methodis_float
(&self)
ug-core/src/dtype.rs:68
← previousnext →301–400 of 526, ranked by callers