deepfold.distributed.utils

Utility functions.

Functions

assert_viewless_tensor(tensor)	Assert that a tensor is not a view.
divide(numerator, denominator)	Ensure that numerator is divisible by the denominator and return the division value.
ensure_divisibility(numerator, denominator)	Ensure that numerator is divisible by the denominator.
safely_set_viewless_tensor_data(tensor, ...)	Safely set tensor's .data field.

Classes

GlobalMemoryBuffer()

Global buffer to avoid dynamic memory allocations.