mirror of
https://github.com/linkedin/Liger-Kernel.git
synced 2026-10-12 02:17:47 +00:00
Vaibhav JindalandSteven Shimizu
389d9cf369
## Summary <!--- This is a required section; please describe the main purpose of this proposed code change. ---> 1. Recent PR (https://github.com/linkedin/Liger-Kernel/pull/876) added functionality to run tests in parallel. However, this leads to GPU OOM errors breaking the CI. Even commands like `pytest test/transformers/test_tvd.py` are not working on a single GPU setup because of parallelism. This PR fixes this issue by changing the behavior to run tests sequentially. 2. Fixes flaky tests for bf16 for glm4v models by increasing tolerance. 3. Use H100 for nvidia tests. <!--- ## Details This is an optional section; is there anything specific that reviewers should be aware of? ---> ## Testing Done <!--- This is a required section; please describe how this change was tested. ---> <!-- Replace BLANK with your device type. For example, A100-80G-PCIe Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> - Hardware Type: <BLANK> - [x] run `make test` to ensure correctness - [x] run `make checkstyle` to ensure code style - [x] run `make test-convergence` to ensure convergence --------- Co-authored-by: Steven Shimizu <[email protected]>