Jagrit Digani 18807aae0b Add fixes to int QMMs (CI passing)

2025-11-19 11:15:52 -08:00

..

jit

Add quantize/dequantize for mxfp8 and nvfp4 (#2688)

2025-10-28 16:23:12 -07:00

kernels

Add fixes to int QMMs (CI passing)

2025-11-19 11:15:52 -08:00

allocator.cpp

fix warnings showing up with -Wall (#2692)

2025-10-24 11:43:35 -07:00

allocator.h

fix warnings showing up with -Wall (#2692)

2025-10-24 11:43:35 -07:00

binary.cpp

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

binary.h

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

CMakeLists.txt

[WIP] Init NAX matmuls

2025-11-19 11:15:52 -08:00

compiled.cpp

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

conv.cpp

faster depthwise 1D conv (#2567)

2025-09-08 11:37:23 -07:00

copy.cpp

[CUDA] Reduce use of managed memory (#2725)

2025-11-05 16:05:23 -08:00

custom_kernel.cpp

export custom kernel (#2756)

2025-11-13 11:29:50 -08:00

device.cpp

[CUDA] Tune ops per buffer based on device (#2761)

2025-11-16 06:29:49 -08:00

device.h

[WIP] Add init QMMs (some CI tests failing)

2025-11-19 11:15:52 -08:00

distributed.cpp

Nccl reduce scatter, all gather (#2727)

2025-11-05 08:21:11 -08:00

eval.cpp

fix warnings showing up with -Wall (#2692)

2025-10-24 11:43:35 -07:00

event.cpp

[Metal] Release metal events (#2412)

2025-07-23 19:53:42 -07:00

fence.cpp

[CUDA] Reduce use of managed memory (#2725)

2025-11-05 16:05:23 -08:00

fft.cpp

[CUDA] Reduce use of managed memory (#2725)

2025-11-05 16:05:23 -08:00

hadamard.cpp

Move common gpu primitives to backend/gpu (#2145)

2025-05-05 13:45:29 -07:00

indexing.cpp

Fix incorrect path and typos (#2630)

2025-09-28 06:03:04 -07:00

jit_kernels.cpp

Add quantize/dequantize for mxfp8 and nvfp4 (#2688)

2025-10-28 16:23:12 -07:00

kernels.h

fix METAL quantization in JIT (#2553)

2025-08-28 18:26:25 -07:00

logsumexp.cpp

Add contiguous_copy_gpu util for copying array (#2379)

2025-07-18 06:44:25 -07:00

make_compiled_preamble.sh

Dispatch bf16 at run time when using the JIT (#1584)

2024-11-15 16:54:36 -08:00

matmul.cpp

Add fixes to int QMMs (CI passing)

2025-11-19 11:15:52 -08:00

matmul.h

Collection of refactors (#2274)

2025-06-13 10:44:56 -07:00

metal.cpp

only build for macos 14 and up (#2731)

2025-11-04 09:44:15 -08:00

metal.h

Generalize gpu backend (#2138)

2025-04-30 09:08:17 -07:00

no_metal.cpp

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

nojit_kernels.cpp

fix METAL quantization in JIT (#2553)

2025-08-28 18:26:25 -07:00

normalization.cpp

fix warnings showing up with -Wall (#2692)

2025-10-24 11:43:35 -07:00

primitives.cpp

fix warnings showing up with -Wall (#2692)

2025-10-24 11:43:35 -07:00

quantized.cpp

Add fixes to int QMMs (CI passing)

2025-11-19 11:15:52 -08:00

reduce.cpp

Fix reduce sum/prod overflow (#2477)

2025-08-12 00:05:33 -07:00

reduce.h

Reductions update (#1351)

2024-11-04 22:25:16 -08:00

resident.cpp

Generalize gpu backend (#2138)

2025-04-30 09:08:17 -07:00

resident.h

Wired (#1510)

2024-10-25 09:35:33 -07:00

rope.cpp

no copy batch rope (#2595)

2025-09-15 14:23:48 -07:00

scaled_dot_product_attention.cpp

[WIP] Add init QMMs (some CI tests failing)

2025-11-19 11:15:52 -08:00

scan.cpp

fix warnings showing up with -Wall (#2692)

2025-10-24 11:43:35 -07:00

slicing.cpp

[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533)

2025-08-26 07:31:39 +09:00

softmax.cpp

Add contiguous_copy_gpu util for copying array (#2379)

2025-07-18 06:44:25 -07:00

sort.cpp

Move common gpu primitives to backend/gpu (#2145)

2025-05-05 13:45:29 -07:00

ternary.cpp

Faster triu, tril, where with scalar (#2644)

2025-10-02 12:21:27 -07:00

ternary.h

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

unary.cpp

Fp8 conversion (#2686)

2025-10-27 16:35:50 -07:00

unary.h

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00

utils.cpp

Move some dims utils to common (#2223)

2025-05-29 06:48:30 -07:00

utils.h

Add Primitive::name and remove Primitive::print (#2365)

2025-07-14 14:06:35 -07:00