SIGN IN SIGN UP

Fix FA3 segfault with custom CUDA streams in ABI stable build (#1957)

The ABI stable implementation incorrectly used getCurrentStream().id()
which returns a StreamId (int64_t) instead of the actual cudaStream_t
pointer. Casting an integer ID to a stream pointer caused segmentation
faults when using custom CUDA streams.

Fixed by using the proper AOTI C API function aoti_torch_get_current_cuda_stream()
which returns the actual CUDA stream pointer.
K
Kevin Wang committed
3effce828cd3c69cdeff96b418a6370d5d5a2430
Parent: e4d25a4
Committed by GitHub <noreply@github.com> on 10/24/2025, 5:17:39 AM