(Mostly internal changes)
Expose a flag to allow local device to own a separate Eigen Threadpool. Enable this flag as a default option in Python benchmarks, so that the number of threads can vary across each unit test run. PiperOrigin-RevId: 237285687
Loading
Please sign in to comment