Build and test

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

Sign in to view logs

Re-run triggered November 27, 2024 18:47

qihqi

#8333

dudulightricks:bug-fix/shard-segment-ids-flash-attention

Status Success

Total duration 1h 28m 37s

Artifacts 2

build_and_test.yml

on: pull_request

get-torch-commit

Build PyTorch/XLA / build

Build docs / build-docs

TPU tests / tpu-test

Matrix: CPU tests / test

Artifacts

Produced during runtime

Name	Size
cpp-test-bin	660 MB
torch-xla-wheels	222 MB