Skip to content

custom_kernel: fix shape mismatch by sharding segment_ids in flash attn. #10436

custom_kernel: fix shape mismatch by sharding segment_ids in flash attn.

custom_kernel: fix shape mismatch by sharding segment_ids in flash attn. #10436

CPU tests  /  test (cpp_tests, cpp_tests2)

succeeded Nov 27, 2024 in 5m 35s