Awni Hannun 66519fb348 fix slice (#2758)

2025-11-13 11:30:02 -08:00

..

binary.h

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

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

broadcasting.cpp

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

2025-11-05 16:05:23 -08: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

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

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

compiled.h

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

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

copy.h

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

2025-11-05 16:05:23 -08: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

Debug cuda conv (#2662)

2025-10-10 16:12:47 -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

fix slice (#2758)

2025-11-13 11:30:02 -08:00

slicing.h

Fix a couple of slicing bugs (#1827)

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

ternary.h

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

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

unary.h

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

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

utils.cpp

[CUDA] Fix conv grads with groups (#2495)

2025-08-16 10:09:18 +09:00

utils.h

[CUDA] Fix conv grads with groups (#2495)

2025-08-16 10:09:18 +09:00