openfold.utils.precision_utils¶

Classes

PrecisionWrapper(model, precision)

Functions

`cast_all`(x, from_dtype, to_dtype)
`cast_tensor`(x, from_dtype, to_dtype)
`is_fp16_enabled`()
`wrap_for_precision`(model, precision)

class PrecisionWrapper(model, precision)¶

Bases: Module

forward(*args, **kwargs)¶

cast_all(x, from_dtype, to_dtype)¶

cast_tensor(x, from_dtype, to_dtype)¶

is_fp16_enabled()¶

wrap_for_precision(model, precision)¶