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