Alexander MatveevandClaude Opus 4.6 c55216ebbf Address review: add CUDA error checking and runtime buffer overflow guard ...

- Add PUSH_AR_CUDACHECK macro wrapping all CUDA API calls (cudaGetDevice,
  cudaDeviceGetAttribute, cudaMalloc, cudaMemset, cudaIpcGetMemHandle,
  cudaIpcOpenMemHandle) to match the CUDACHECK pattern in custom_all_reduce.cuh
- Replace assert(input_bytes <= push_buffer_bytes_) with a runtime
  std::runtime_error check that is not compiled out under -DNDEBUG
- Add #include <stdexcept> and #include <string> for the runtime check

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Signed-off-by: Alexander Matveev <alexm-redhat@dgx-b200-02.mgmt.accl-001.lab.rdu2.dc.redhat.com>

2026-06-15 18:09:24 -04:00

..

attention

[9/n] Migrate attention and cache kernels to torch stable ABI (continued) (#43717)

2026-05-29 04:44:45 +00:00

core

[5/n] Migrate CUTLASS MLA, hadamard, awq, allspark and DSV3 fused a gemm to torch stable ABI (continued) (#42339)

2026-05-13 07:24:39 +00:00

cpu

[CPU] Refine CPU attention frontend (#45391)

2026-06-14 19:26:54 -07:00

cutlass_extensions

[11b/n] Migrate Machete kernels to torch stable ABI (#45304)

2026-06-12 10:36:49 +00:00

libtorch_stable

[Model] Add MiniMax M3 support (#45381)

2026-06-16 01:01:25 +08:00

moe

[10c/n] Migrate MoE kernels to torch stable ABI (#44565)

2026-06-10 23:02:26 -07:00

quantization

[11b/n] Migrate Machete kernels to torch stable ABI (#45304)

2026-06-12 10:36:49 +00:00

quickreduce

[Bugfix][Rocm] fix qr error when different inp shape (#25892)

2025-10-13 10:04:21 -07:00

rocm

[ROCm][Perf] Fused MoE W4A16 HIP kernel for AMD RDNA3 (gfx1100) (#44075)

2026-06-06 15:30:39 -05:00

cache.h

Use CU_MEMCPY_SRC_ACCESS_ORDER_ANY for batch KV cache swaps (#39306)

2026-05-10 05:57:09 +03:00

cub_helpers.h

[Refactor] Refactor FP8 & INT8 Quant Folder inside w8a8 (#25293)

2025-10-08 10:20:48 -04:00

cuda_compat.h

[Bugfix][ROCm] Fix for warp_size uses on host (#21205)

2025-07-24 00:37:19 -07:00

cuda_utils.h

[Attention] MLA with chunked prefill (#12639)

2025-02-21 15:30:12 -08:00

cuda_view.cu

[Bugfix] [ROCm] [Critical] fallback to regular abi for ROCm (#44648)

2026-06-05 15:51:17 +00:00

cumem_allocator_compat.h

[Core][AMD] Migrate fully transparent sleep mode to ROCm platform (#12695)

2025-11-12 15:24:12 -08:00

cumem_allocator.cpp

[ROCm][CI] Stabilize sleep-mode memory release (#43022)

2026-06-09 16:51:12 +08:00

custom_all_reduce_test.cu

[Distributed] Add custom allreduce support for ROCM (#14125)

2025-03-31 22:49:12 -07:00

custom_all_reduce.cuh

[Kernels] Enable Torch Symmetric Memory All-Reduce By Default (#24111)

2025-09-11 09:45:31 -07:00

custom_quickreduce.cu

[Feature] add quick all reduce (#19744)

2025-06-26 20:54:24 -07:00

dispatch_utils.h

[Performance] Fused blockwise quant RMS norm (#27883)

2025-12-07 16:38:04 +00:00

minimax_reduce_rms_kernel.h

[10b/n] Migrate custom all-reduce, DeepSeek V4 fused MLA, MiniMax reduce-RMS, and MXFP8 MoE to libtorch stable ABI (#44365)

2026-06-04 00:29:46 +08:00

ops.h

[Model] Add MiniMax M3 support (#45381)

2026-06-16 01:01:25 +08:00

push_all_reduce_kernel.cuh

perf: add push-based allreduce for small tensor reductions

2026-06-15 18:09:24 -04:00

push_all_reduce.cu

perf: add push-based allreduce for small tensor reductions

2026-06-15 18:09:24 -04:00

push_all_reduce.cuh

Address review: add CUDA error checking and runtime buffer overflow guard

2026-06-15 18:09:24 -04:00

spinloop.cpp

Add VLLM_USE_SPINLOOP_EXT to use more efficient busy polling (#36517)

2026-05-11 16:11:49 -07:00

torch_bindings.cpp

perf: add push-based allreduce for small tensor reductions

2026-06-15 18:09:24 -04:00

torch_utils.h

[6/n] Migrate activation kernels, gptq, gguf, non cutlass w8a8 to libtorch stable ABI (continued) (#42663)

2026-05-20 00:18:12 -07:00

type_convert.cuh

[10b/n] Migrate custom all-reduce, DeepSeek V4 fused MLA, MiniMax reduce-RMS, and MXFP8 MoE to libtorch stable ABI (#44365)

2026-06-04 00:29:46 +08:00