MCPcopy Create free account

hub / github.com/LaurentMazare/ug / functions

Functions526 in github.com/LaurentMazare/ug

⨍ Functions 526◇ Types & classes 143
Methodis_fortran_contiguous
Returns true if the strides are Fortran contiguous (aka column major).
ug-core/src/layout.rs:177
Methodis_int
(&self)
ug-core/src/dtype.rs:61
Functionlb_copy
()
ug-core/tests/lb_tests.rs:67
Functionlb_cuda_add
()
ug-cuda/tests/lb_tests.rs:4
Functionlb_cuda_cat
()
ug-cuda/tests/lb_tests.rs:16
Functionlb_cuda_mm
()
ug-cuda/tests/lb_tests.rs:10
Functionlb_custom
()
ug-core/tests/lb_tests.rs:112
Functionlb_custom_in_place
()
ug-core/tests/lb_tests.rs:174
Functionlb_layout
()
ug-core/tests/lb_tests.rs:153
Functionlb_metal_add
()
ug-metal/tests/lb_tests.rs:4
Functionlb_metal_cat
()
ug-metal/tests/lb_tests.rs:16
Functionlb_metal_mm
()
ug-metal/tests/lb_tests.rs:10
Functionlb_set
()
ug-core/tests/lb_tests.rs:207
Methodload
(&self, name: &str, device: &D)
ug-core/src/safetensors.rs:68
Methodload
(src: usize, offset: usize, dtype: DType)
ug-pyo3/src/lib.rs:319
Functionlower_add
()
ug-core/tests/basic_tests.rs:28
Functionmain
()
ug-metal/examples/metal_bench.rs:120
Functionmain
()
ug-core/examples/dot_prod.rs:49
Functionmain
()
ug-core/examples/basics.rs:5
Functionmain
()
ug-llama/src/main.rs:65
Functionmain
()
ug-cuda/examples/benchmark.rs:105
Functionmain
()
ug-cuda/examples/cuda_gen.rs:79
Methodmatmul
( &self, dst: &mut Self::Slice, lhs: &Self::Slice, rhs: &Self::Slice,
ug-metal/src/runtime.rs:128
Methodmatmul
( &self, dst: &mut Self::Slice, lhs: &Self::Slice, rhs: &Self::Slice,
ug-core/src/cpu_runtime.rs:336
Methodmatmul
( &self, dst: &mut Self::Slice, lhs: &Self::Slice, rhs: &Self::Slice,
ug-cuda/src/runtime.rs:334
Methodmax_value
(dtype: DType)
ug-core/src/const.rs:165
Functionmember_sort
(member)
stub.py:58
Methodmergeable_adjacent_dims
(&self)
ug-core/src/lower_op.rs:666
Methodmin
(&self, axis: usize)
ug-pyo3/src/lib.rs:189
Methodmin_value
(dtype: DType)
ug-core/src/const.rs:155
Functionmod_
(py: Python, m: &Bound<PyModule>)
ug-pyo3/src/lib.rs:602
Methodmsg
(err: impl std::error::Error)
ug-core/src/error.rs:61
Methodmul
(&mut self, src_id: Id, v: i32)
ug-core/src/block.rs:86
Functionmul_cst
(lb: &LB<D>, v: f32)
ug-core/tests/display_tests.rs:3
Methodmulti
Creates a wrapper around multiple memory mapped file and deserialize the safetensors headers. If a tensor name appears in multiple files, the last en
ug-core/src/safetensors.rs:44
Methodname
(&self)
ug-core/src/lower_op.rs:561
Methodname
(&self)
ug-core/src/cpu_runtime.rs:437
Methodnew
()
ug-metal/src/runtime.rs:24
Methodnew
(f: F)
ug-core/src/lazy_buffer.rs:28
Methodnew
()
ug-core/src/display.rs:136
Methodnew
Creates a wrapper around a memory mapped file and deserialize the safetensors header. # Safety The unsafe is inherited from [`memmap2::MmapOptions`]
ug-core/src/safetensors.rs:21
Methodnew
()
ug-core/src/lang.rs:88
Methodnew
()
ug-core/src/cpu_runtime.rs:419
Methodnew
(id_cnts: HashMap<crate::lazy_buffer::Id, usize>)
ug-core/src/schedule.rs:316
Methodnew
(k: &op::Kernel)
ug-core/src/cache.rs:15
Methodnew
(v: usize)
ug-core/src/interpreter.rs:11
Methodnew
()
ug-core/src/block.rs:11
Methodnew
(dim: usize, eps: f64, name: &str, st: &ST, device: &D)
ug-llama/src/model.rs:117
Methodnew
()
ug-cuda/src/runtime.rs:37
Methodnew
(dst: Arg, shape: Vec<usize>, value: Ast)
ug-pyo3/src/lib.rs:233
Methodnull
A placeholder value, never to be used for anything else.
ug-core/src/lang.rs:631
Methodpartial_cmp
(&self, other: &Self)
ug-core/src/lang.rs:145
Functionprint_options
()
ug-core/src/display.rs:79
Methodprogram_id
()
ug-pyo3/src/lib.rs:533
Methodptr_bf16
()
ug-pyo3/src/lib.rs:109
Methodptr_f16
()
ug-pyo3/src/lib.rs:114
Methodptr_i32
()
ug-pyo3/src/lib.rs:124
Methodrange
(lo: usize, up: usize, end_idx: usize)
ug-pyo3/src/lib.rs:313
Methodrope
(src: &LB<Self>, cos: &LB<Self>, sin: &LB<Self>, pos: &LB<Self>)
ug-llama/src/cpu_ops.rs:91
Methodrope
(src: &LB<Self>, cos: &LB<Self>, sin: &LB<Self>, pos: &LB<Self>)
ug-llama/src/metal_ops.rs:54
Methodrope
(src: &LB<Self>, cos: &LB<Self>, sin: &LB<Self>, pos: &LB<Self>)
ug-llama/src/cuda_ops.rs:58
Methodrope_i
(src: &LB<Self>, cos: &LB<Self>, sin: &LB<Self>, pos: &LB<Self>)
ug-llama/src/cpu_ops.rs:58
Methodrope_i
(src: &LB<Self>, cos: &LB<Self>, sin: &LB<Self>, pos: &LB<Self>)
ug-llama/src/metal_ops.rs:12
Methodrope_i
(src: &LB<Self>, cos: &LB<Self>, sin: &LB<Self>, pos: &LB<Self>)
ug-llama/src/cuda_ops.rs:18
Methodrun
(&self, f: &Self::Func, args: &mut [&mut Self::Slice])
ug-metal/src/runtime.rs:108
Methodrun
(&self, f: &Self::Func, args: &mut [&mut Self::Slice])
ug-core/src/cpu_runtime.rs:288
Methodrun0
(&self)
ug-core/src/cpu_runtime.rs:442
Functionrun_add
()
ug-core/tests/cpu_tests.rs:4
Functionschedule_broadcast
()
ug-core/tests/lb_tests.rs:100
Functionschedule_cat
()
ug-core/tests/lb_tests.rs:62
Functionschedule_cpu
()
ug-core/tests/lb_tests.rs:30
Functionschedule_cpu_compile
()
ug-core/tests/lb_tests.rs:52
Functionschedule_interpret
()
ug-core/tests/lb_tests.rs:5
Functionschedule_mm
()
ug-core/tests/lb_tests.rs:57
Functionset_edge_items
(edge_items: usize)
ug-core/src/display.rs:119
Functionset_gemm_reduced_precision_bf16
This bool controls whether reduced precision reductions (e.g., with fp16 accumulation type) are allowed with bf16 GEMMs.
ug-cuda/src/gemm.rs:126
Functionset_gemm_reduced_precision_f16
This bool controls whether reduced precision reductions (e.g., with fp16 accumulation type) are allowed with f16 GEMMs.
ug-cuda/src/gemm.rs:114
Functionset_gemm_reduced_precision_f32
This bool controls whether reduced precision reductions (e.g., with tf32 accumulation type) are allowed with f32 GEMMs.
ug-cuda/src/gemm.rs:102
Functionset_line_width
(line_width: usize)
ug-core/src/display.rs:111
Methodset_param
(encoder: &ComputeCommandEncoderRef, position: u64, data: Self)
ug-metal/src/utils.rs:36
Functionset_precision
(precision: usize)
ug-core/src/display.rs:115
Functionset_print_options
(options: PrinterOptions)
ug-core/src/display.rs:83
Functionset_print_options_default
()
ug-core/src/display.rs:87
Functionset_print_options_full
()
ug-core/src/display.rs:101
Functionset_print_options_short
()
ug-core/src/display.rs:91
Functionset_sci_mode
(sci_mode: Option<bool>)
ug-core/src/display.rs:127
Functionset_threshold
(threshold: usize)
ug-core/src/display.rs:123
Methodshape
(&self)
ug-core/src/layout.rs:640
Methodshape
(&self, py: Python)
ug-pyo3/src/lib.rs:199
Methodsize_in_bytes
The size used by each element in bytes, i.e. 4 for `F32`.
ug-core/src/dtype.rs:51
Methodslice
(&self, vs: Vec<f32>)
ug-pyo3/src/lib.rs:464
Methodspecial_bi
()
ug-pyo3/src/lib.rs:344
Methodspecial_ti
()
ug-pyo3/src/lib.rs:339
Methodsplat
(v: T)
ug-core/src/interpreter.rs:93
Methodssa
( ssa: crate::lang::ssa::Kernel, args: Vec<Self>, s: S, dtype: DType,
ug-core/src/lazy_buffer.rs:387
Functionssa_softmax
()
ug-core/tests/basic_tests.rs:65
Functionssa_softmax_reduce
()
ug-core/tests/basic_tests.rs:92
Methodstore
(dst: usize, offset: usize, value: usize, dtype: DType)
ug-pyo3/src/lib.rs:324
Functionstride
()
ug-core/src/layout.rs:564
Methodsynchronize
(&self)
ug-metal/src/runtime.rs:170
← previousnext →401–500 of 526, ranked by callers