Cheng 4822c3dbe9 [CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533) ...

* Move DynamicSlice to gpu/primitives

* Implement compute_dynamic_offset in CUDA

2025-08-26 07:31:39 +09:00

..

binary

Faster general unary op (#2472)

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

conv

[CUDA] Add GEMM-based fallback convolution kernels (#2511)

2025-08-20 10:06:22 +09:00

copy

Faster general unary op (#2472)

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

device

[CUDA] Remove thrust in arange (#2535)

2025-08-24 16:22:36 +09:00

gemms

[CUDA] Add GEMM-based fallback convolution kernels (#2511)

2025-08-20 10:06:22 +09:00

quantized

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

reduce

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

steel

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

unary

Faster general unary op (#2472)

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

allocator.cpp

[CUDA] Update calls to cudaMemAdvise and cudaGraphAddDependencies for CUDA 13 (#2525)

2025-08-21 19:57:20 -07:00

allocator.h

[CUDA] Simplify allocator (#2392)

2025-07-22 08:24:01 -07:00

arange.cu

[CUDA] Remove thrust in arange (#2535)

2025-08-24 16:22:36 +09:00

arg_reduce.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

bin2h.cmake

CUDA backend: compile (#2276)

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

binary_two.cu

Faster general unary op (#2472)

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

CMakeLists.txt

NCCL backend (#2476)

2025-08-21 11:56:15 -07:00

compiled.cpp

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

conv.cpp

[CUDA] Add GEMM-based fallback convolution kernels (#2511)

2025-08-20 10:06:22 +09:00

copy.cu

[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533)

2025-08-26 07:31:39 +09:00

cuda.cpp

start cuda circle config (#2256)

2025-06-10 21:19:47 -07:00

cuda.h

start cuda circle config (#2256)

2025-06-10 21:19:47 -07:00

cudnn_utils.cpp

[CUDA] Fix stride of singleton dims before passing to cuDNN (#2521)

2025-08-21 08:55:26 +09:00

cudnn_utils.h

Split cuDNN helpers into a separate header (#2491)

2025-08-20 09:29:28 +09:00

custom_kernel.cpp

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

device.cpp

[CUDA] Update calls to cudaMemAdvise and cudaGraphAddDependencies for CUDA 13 (#2525)

2025-08-21 19:57:20 -07:00

device.h

Split cuDNN helpers into a separate header (#2491)

2025-08-20 09:29:28 +09:00

distributed.cu

Fix allocation bug in NCCL (#2530)

2025-08-22 14:39:43 -07:00

eval.cpp

[CUDA] Save primitive inputs faster (#2449)

2025-08-01 10:16:06 +09:00

event.cu

[CUDA] Simplify allocator (#2392)

2025-07-22 08:24:01 -07:00

event.h

[CUDA] Simplify allocator (#2392)

2025-07-22 08:24:01 -07:00

fence.cpp

Avoid atomic updates across CPU/GPU in CUDA event (#2231)

2025-06-03 16:49:06 -07:00

indexing.cpp

[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533)

2025-08-26 07:31:39 +09:00

jit_module.cpp

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

jit_module.h

[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533)

2025-08-26 07:31:39 +09:00

kernel_utils.cu

Remove the kernel arg from get_launch_args (#2437)

2025-07-30 11:43:02 +09:00

kernel_utils.cuh

Use SmallVector for shapes and strides (#2454)

2025-08-05 09:41:03 +09:00

layer_norm.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

logsumexp.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

lru_cache.h

Use LRU cache for cuda graph (#2448)

2025-08-02 21:28:57 +09:00

matmul.cpp

Rename cu::Matmul to CublasGemm (#2488)

2025-08-13 09:37:40 +09:00

no_cuda.cpp

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

primitives.cpp

[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533)

2025-08-26 07:31:39 +09:00

random.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

reduce.cu

faster rms norm (#2433)

2025-07-29 13:12:00 -07:00

rms_norm.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

rope.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

scaled_dot_product_attention.cu

Add CUDA sdpa vector (#2468)

2025-08-06 21:40:26 -07:00

scan.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

slicing.cpp

[CUDA] Implement DynamicSlice/DynamicSliceUpdate (#2533)

2025-08-26 07:31:39 +09:00

softmax.cu

[CUDA] Matmul utils initial commit (#2441)

2025-08-01 14:22:25 -07:00

sort.cu

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

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

ternary.cu

Faster general unary op (#2472)

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

unary.cu

Faster general unary op (#2472)

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

utils.cpp

Split cuDNN helpers into a separate header (#2491)

2025-08-20 09:29:28 +09:00

utils.h

Split cuDNN helpers into a separate header (#2491)

2025-08-20 09:29:28 +09:00

worker.cpp

[CUDA] Simplify allocator (#2392)

2025-07-22 08:24:01 -07:00

worker.h

[CUDA] Simplify allocator (#2392)

2025-07-22 08:24:01 -07:00