Skip to content

v1.1.1

Latest

Choose a tag to compare

@AKJ7 AKJ7 released this 15 Oct 19:16
83ec427

1.1.1

A fix for --torch-num-threads=auto under pytest-xdist: it now leaves a thread count you
set with MKL_NUM_THREADS alone, as it already did for OMP_NUM_THREADS.

Fixed

  • --torch-num-threads=auto also leaves the thread count alone when MKL_NUM_THREADS is set,
    not only OMP_NUM_THREADS: torch gives MKL_NUM_THREADS precedence, so it's your setting
    too. An explicit count now sets both variables for subprocesses.
  • The test of that behavior no longer depends on how torch resolves the two variables on
    the machine running it; it failed on CI runners that set MKL_NUM_THREADS.