[CI][RFC] Replace black-jupyter with ruff-format (#37210)

Co-authored-by: Alison Shao <a.shao@wustl.edu>
This commit is contained in:
Alex Nails
2026-09-02 19:46:08 -07:00
committed by GitHub
co-authored by Alison Shao
parent 2641e427be
commit 28262c20df
1411 changed files with 7766 additions and 8176 deletions
@@ -62,9 +62,7 @@ def _run_rank(rank, world_size, port, scenario, result_q):
free, _total = torch.cuda.mem_get_info(rank)
target = max(free - (1 << 30), 0)
granularity_flag = (
cuda_driver.CUmemAllocationGranularity_flags.CU_MEM_ALLOC_GRANULARITY_RECOMMENDED
)
granularity_flag = cuda_driver.CUmemAllocationGranularity_flags.CU_MEM_ALLOC_GRANULARITY_RECOMMENDED
err, gran = cuda_driver.cuMemGetAllocationGranularity(
prop,
granularity_flag,