CastConfig¶
- class torchao.float8.CastConfig(scaling_type: ScalingType = ScalingType.DYNAMIC, scaling_granularity: ScalingGranularity = ScalingGranularity.TENSORWISE, target_dtype: Optional[dtype] = None)[源代码]¶
用于可能将单个张量转换为 float8 的配置
用于可能将单个张量转换为 float8 的配置