Closed nestordemeure closed 1 year ago
This branch introduces a number of modifications that were useful for benchmarking.
In particular:
TOAST_FUNCTIME
jax_accel_assign_device
This branch introduces a number of modifications that were useful for benchmarking.
In particular:
TOAST_FUNCTIME
read its value instead of considering any value true (this is useful when switching from true/1 to false/0 inside a script),jax_accel_assign_device
function to deal properly with gpu oversubscription (several process using the same gpu)