deepfold.testing.common

Functions

skip_if_lt_x_gpu(n)
skip_if_no_gpu(func)	Skips if the world size exceeds the number of GPUs.
with_nccl_blocking_wait(func)	Convenience decorator to set/unset TORCH_NCCL_BLOCKING_WAIT flag.

Classes

MultiProcessTestCase([meethod_name])