mirror of
https://github.com/linkedin/Liger-Kernel.git
synced 2026-10-12 02:17:47 +00:00
Automate benchmarking (#760)
## Summary <!--- This is a required section; please describe the main purpose of this proposed code change. ---> This PR sets the time limit for the modal to run to 90 minutes and also the workflow would be triggered every Friday. <!--- ## 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> - [ ] run `make test` to ensure correctness - [ ] run `make checkstyle` to ensure code style - [ ] run `make test-convergence` to ensure convergence --------- Co-authored-by: Manan Shah <[email protected]> Co-authored-by: Vaibhav Jindal <[email protected]>
This commit is contained in:
co-authored by
Manan Shah
Vaibhav Jindal
parent
1f640a505a
commit
c1bdc24546
@@ -1,13 +1,10 @@
|
||||
name: Benchmarks
|
||||
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- main
|
||||
# paths:
|
||||
# - "src/**"
|
||||
# - "test/**"
|
||||
# - "benchmark/**"
|
||||
schedule:
|
||||
# Runs at 00:00 UTC every Friday
|
||||
- cron: '0 0 * * 5'
|
||||
workflow_dispatch: # Enables manual trigger
|
||||
|
||||
permissions:
|
||||
contents: write
|
||||
|
||||
@@ -14,7 +14,7 @@ app = modal.App("liger_benchmarks", image=image)
|
||||
repo = image.add_local_dir(ROOT_PATH, remote_path=REMOTE_ROOT_PATH)
|
||||
|
||||
|
||||
@app.function(gpu="H100", image=repo, timeout=60 * 45)
|
||||
@app.function(gpu="H100", image=repo, timeout=60 * 90)
|
||||
def liger_benchmarks():
|
||||
import os
|
||||
import subprocess
|
||||
|
||||
Reference in New Issue
Block a user