mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
24 lines
525 B
Python
24 lines
525 B
Python
|
import subprocess
|
||
|
|
from os.path import join
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
script_dir = join(
|
||
|
|
Path(__file__).parent, "experiment_scripts"
|
||
|
|
)
|
||
|
|
encoding = 'utf-8'
|
||
|
|
|
||
|
|
|
||
|
|
def run_script(script_name):
|
||
|
|
# runs the script and asserts that there are no errors
|
||
|
|
p = subprocess.run(
|
||
|
|
["bash", f"{join(script_dir,script_name)}"],
|
||
|
|
stdout=subprocess.PIPE,
|
||
|
|
stderr=subprocess.PIPE
|
||
|
|
)
|
||
|
|
error = p.stderr.decode(encoding)
|
||
|
|
assert not error
|
||
|
|
|
||
|
|
|
||
|
|
def test_ddp_nccl_allreduce():
|
||
|
|
run_script("ddp_nccl_allreduce.sh")
|