tqchen opened a new pull request, #19942: URL: https://github.com/apache/tvm/pull/19942
Add tvm.testing.run_with_gpu_lock backed by the existing tvm_ffi.utils.FileLock. Migrate live local GPU tests to acquire the machine-local lock around device execution, synchronization, host transfer, and checks while leaving target construction and compilation outside the critical section. Replace the custom xdist scheduler with standard xdist_group placement for the order-dependent test family. RPC tests retain dynamic port allocation and per-test process isolation rather than gaining a broad category lock. Validation: all changed Python files pass pre-commit and bytecode compilation. On one H200, the 70-file changed-test selection with --dist=loadgroup scaled from 1446 seconds at one worker to 196 seconds at 16 workers; 24 workers took 208 seconds. No lock timeout, deadlock, or worker loss occurred, and the known baseline outcomes reproduced across worker counts. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
