Awni Hannun dad1b00b13 fix (#1523)

2024-10-24 19:17:46 -07:00

..

jit

Working 64-bit scans (#1506)

2024-10-24 11:05:46 -07:00

kernels

Working 64-bit scans (#1506)

2024-10-24 11:05:46 -07:00

allocator.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

allocator.h

Allow querying the allocator for the buffer size (#1404)

2024-09-11 21:02:16 -07:00

binary.cpp

Faster metal compiled kernels + some fixes (#1486)

2024-10-14 12:45:38 -07:00

binary.h

Fixes for large arrays with a few ops (#1299)

2024-07-30 17:18:39 -07:00

CMakeLists.txt

Chore: add pre-commit hook for cmake (#1362)

2024-09-16 12:53:01 -07:00

compiled.cpp

Fix compile (#1501)

2024-10-18 11:06:40 -07:00

conv.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

copy.cpp

Faster metal compiled kernels + some fixes (#1486)

2024-10-14 12:45:38 -07:00

copy.h

Fix copying scalars by adding fill_gpu (#1402)

2024-09-09 15:54:08 -07:00

custom_kernel.cpp

Remove use of vector<const T> (#1514)

2024-10-22 16:31:52 -07:00

device.cpp

fix (#1523)

2024-10-24 19:17:46 -07:00

device.h

fix (#1523)

2024-10-24 19:17:46 -07:00

distributed.cpp

Adds send/recv ops in distributed (#1366)

2024-08-26 23:01:37 -07:00

event.cpp

Fix array is_available race cases (#1468)

2024-10-07 19:13:50 -07:00

fft.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

hadamard.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

indexing.cpp

C++20 compatibility for fmt (#1519)

2024-10-24 08:54:51 -07:00

jit_kernels.cpp

Working 64-bit scans (#1506)

2024-10-24 11:05:46 -07:00

kernels.h

C++20 compatibility for fmt (#1519)

2024-10-24 08:54:51 -07:00

make_compiled_preamble.sh

fix compiling with space in paths (#1332)

2024-08-15 16:39:24 -07:00

matmul.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

matmul.h

Conv grad with groups + bugfix (#1449)

2024-10-06 07:08:53 -07:00

metal_impl.h

Add synchronize function (#1006)

2024-04-22 08:25:46 -07:00

metal.cpp

Fix array is_available race cases (#1468)

2024-10-07 19:13:50 -07:00

metal.h

Reset peak memory (#1074)

2024-05-03 17:12:51 -07:00

nojit_kernels.cpp

Real and Imag (#1490)

2024-10-15 16:23:15 -07:00

normalization.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

primitives.cpp

Eigenvalues and eigenvectors (#1334)

2024-10-22 12:18:48 -07:00

quantized.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

reduce.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

reduce.h

Further reduction tuning (#1349)

2024-08-23 10:35:25 -07:00

rope.cpp

Xcode 160 (#1384)

2024-09-10 15:15:17 -07:00

scaled_dot_product_attention.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

scan.cpp

Working 64-bit scans (#1506)

2024-10-24 11:05:46 -07:00

slicing.cpp

Fix copying scalars by adding fill_gpu (#1402)

2024-09-09 15:54:08 -07:00

slicing.h

Fix slice data size (#1394)

2024-09-04 19:10:43 -07:00

softmax.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

sort.cpp

Remove Hazard tracking with Fences (#1509)

2024-10-21 19:33:32 -07:00

ternary.cpp

Faster metal compiled kernels + some fixes (#1486)

2024-10-14 12:45:38 -07:00

ternary.h

Add some internal GPU apis (#1177)

2024-06-04 09:24:26 -07:00

unary.cpp

Real and Imag (#1490)

2024-10-15 16:23:15 -07:00

unary.h

Add some internal GPU apis (#1177)

2024-06-04 09:24:26 -07:00

utils.cpp

Working 64-bit scans (#1506)

2024-10-24 11:05:46 -07:00

utils.h

Working 64-bit scans (#1506)

2024-10-24 11:05:46 -07:00