github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix create_schedule_with_vars usage in allreduce benchmark [pr] (#8522)

Browse Source 
* fix create_schedule_with_vars usage in allreduce benchmark [pr]

because i didn't know how to use it...

* increase time limit because tiny17 is slow
This commit is contained in:
chenyu
2025-01-07 01:30:01 -05:00
committed by GitHub
parent 0061dc7447
commit 85a4397f27
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/benchmark.yml vendored
View File

@@ -136,7 +136,7 @@ jobs:
testnvidiabenchmark:
name: tinybox green Benchmark
runs-on: [self-hosted, Linux, tinyboxgreen]
timeout-minutes: 20
timeout-minutes: 30
defaults:
run:
shell: bash -o pipefail {0}

3
test/external/external_benchmark_multitensor_allreduce.py vendored
View File

@@ -9,7 +9,8 @@ from typing import List, Union
def realize(x: Union[UOp, List[UOp]]):
x = x if isinstance(x, list) else [x]
run_schedule(*create_schedule_with_vars(x))
schedule, var_vals, _ = create_schedule_with_vars(x)
run_schedule(schedule, var_vals)
for lb in x: Device[lb.device].synchronize()
def test(devs: List[str], N: int, iters:int = 10):