Cheng 828c5f1137 Use SmallVector for shapes and strides (#2454) ...

* Use SmallVector for shapes and strides

* Convert SmallVector to tuple

2025-08-05 09:41:03 +09:00

..

binary.h

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

broadcasting.cpp

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

broadcasting.h

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

buffer_cache.h

Add memory cache to CUDA backend (#2221)

2025-05-30 12:12:54 -07:00

CMakeLists.txt

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

common.cpp

Gather mm new kernel and small refactoring (#2040)

2025-04-14 16:37:36 -07:00

compiled.cpp

Compile float64 functions on CPU (#2311)

2025-06-24 10:18:52 -07:00

compiled.h

Compile float64 functions on CPU (#2311)

2025-06-24 10:18:52 -07:00

copy.h

CUDA backend: unary ops (#2158)

2025-06-09 06:45:08 -07:00

hadamard.h

GPU Hadamard for large N (#1879)

2025-05-01 17:19:17 -07:00

load.cpp

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

matmul.h

[CUDA] Switch to CUDA graphs (#2317)

2025-07-02 15:59:13 -07:00

reduce.cpp

[CUDA] Fix reductions (#2314)

2025-06-27 12:59:20 -07:00

reduce.h

[CUDA] Fix reductions (#2314)

2025-06-27 12:59:20 -07:00

slicing.cpp

redesign for faster cpu/gpu synch (#1869)

2025-03-06 19:23:38 -08:00

slicing.h

Fix a couple of slicing bugs (#1827)

2025-02-05 19:50:08 -08:00

ternary.h

fix malloc or wait deadlock (#1976)

2025-03-20 16:48:43 -07:00

unary.h

CUDA backend: unary ops (#2158)

2025-06-09 06:45:08 -07:00

utils.cpp

[CUDA] Backward convolution (#2431)

2025-08-01 09:54:05 +09:00

utils.h

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00