Awni Hannun 6441c21a94 Faster general unary op (#2472) ...

* faster general unary op

* faster general ops + reorg

* fix + comment

* binary two

* copy general

2025-08-15 15:04:12 -07:00

..

atomic_ops.cuh

Add more CUDA architectures for PyPi package (#2427)

2025-07-28 12:35:15 -07:00

binary_ops.cuh

[CUDA] Use cuda::std::complex in place of cuComplex (#2372)

2025-07-15 00:36:13 -07:00

cast_op.cuh

[CUDA] Use cuda::std::complex in place of cuComplex (#2372)

2025-07-15 00:36:13 -07:00

complex.cuh

fix complex reduce + nan propagation in min and max (#2377)

2025-07-15 18:19:47 -07:00

config.h

divmod, partition, sort fixes (#2302)

2025-06-16 18:49:32 -07:00

fp16_math.cuh

CUDA backend: compile (#2276)

2025-06-12 17:08:39 -07:00

gather_axis.cuh

CUDA backend: indexing ops (#2277)

2025-06-12 21:44:19 -07:00

gather.cuh

CUDA backend: indexing ops (#2277)

2025-06-12 21:44:19 -07:00

indexing.cuh

CUDA backend: indexing ops (#2277)

2025-06-12 21:44:19 -07:00

scatter_axis.cuh

CUDA backend: indexing ops (#2277)

2025-06-12 21:44:19 -07:00

scatter_ops.cuh

CUDA backend: indexing ops (#2277)

2025-06-12 21:44:19 -07:00

scatter.cuh

CUDA backend: indexing ops (#2277)

2025-06-12 21:44:19 -07:00

ternary_ops.cuh

Cuda bug fixes 2 (#2298)

2025-06-16 13:14:46 -07:00

unary_ops.cuh

[CUDA] Use cuda::std::complex in place of cuComplex (#2372)

2025-07-15 00:36:13 -07:00

utils.cuh

Faster general unary op (#2472)

2025-08-15 15:04:12 -07:00