Awni Hannun 0eb56d5be0 Wired (#1510) ...

* expose residency sets as wire/unwire

* returns wired size

* fix

* runtime support check

* fix os check

* fix test

* fix no metal build

* docs

* nit

* nits in docs

* nits

2024-10-25 09:35:33 -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

Wired (#1510)

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

allocator.h

Wired (#1510)

2024-10-25 09:35:33 -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

Wired (#1510)

2024-10-25 09:35:33 -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

Wired (#1510)

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

device.h

Wired (#1510)

2024-10-25 09:35:33 -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

Wired (#1510)

2024-10-25 09:35:33 -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

Wired (#1510)

2024-10-25 09:35:33 -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

resident.cpp

Wired (#1510)

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

resident.h

Wired (#1510)

2024-10-25 09:35:33 -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