jax
0da52cd1 - [Mosaic GPU] Skip OSS Flash Attention test unless running on sm90

Commit

1 year ago

[Mosaic GPU] Skip OSS Flash Attention test unless running on sm90

References

#24322 - [Mosaic GPU] Skip OSS Flash Attention test unless running on sm90

Author

andportnoy

andportnoy

Parents

Loading