search

pytorch / builder

Continuous builder and binary build scripts for pytorch
BSD 2-Clause "Simplified" License
323 stars 213 forks source link

Add smoke test for CUDA_VISIBLE_DEVICES #1906

Closed atalman closed 4 days ago

atalman commented 4 days ago

Add smoke test for : https://github.com/pytorch/pytorch/issues/128819

malfet commented 4 days ago

Calling CUDA_VISIBLE_DEVICES after first allocation is done would not have any effect on the system, so it does not test for the regression, is it?