Ivar Flakstad e63bb8661b Merge branch 'main' into ivarflakstad/metal-prng

2024-01-12 07:19:58 +01:00

..

affine.metal

Use __HAVE_BFLOAT__ to check for bfloat support instead of metal version check (#1540)

2024-01-10 18:50:30 +01:00

binary.metal

Use __HAVE_BFLOAT__ to check for bfloat support instead of metal version check (#1540)

2024-01-10 18:50:30 +01:00

cast.metal

remove metal version check

2024-01-11 21:02:03 +00:00

conv.metal

Adding upsample_nearest_2d.

2023-12-25 14:25:19 +01:00

indexing.metal

feat(bf16): add cast support + tests for cast + bin ops (#1524)

2024-01-11 15:49:13 +01:00

lib.rs

Merge branch 'main' into ivarflakstad/metal-prng

2024-01-12 07:19:58 +01:00

libMetalFlashAttention.metallib

Removed MPSMatrix entirely (buggy).

2023-12-13 16:21:48 +01:00

random.metal

Gaussian normal distribution of PRNG via Box-Muller transform

2024-01-07 11:39:46 +01:00

reduce.metal

Use __HAVE_BFLOAT__ to check for bfloat support instead of metal version check (#1540)

2024-01-10 18:50:30 +01:00

ternary.metal

Metal: more u8/u32 (#1502)

2023-12-29 23:56:21 +01:00

tests.rs

Merge branch 'main' into ivarflakstad/metal-prng

2024-01-12 07:19:58 +01:00

unary.metal

Use __HAVE_BFLOAT__ to check for bfloat support instead of metal version check (#1540)

2024-01-10 18:50:30 +01:00