Function tvm::tirx::transform::FP8ComputeLegalize

Function tvm::tirx::transform::FP8ComputeLegalize#

Function Documentation#

Pass tvm::tirx::transform::FP8ComputeLegalize(ffi::String promote_dtype = "float16")#

Legalize fp8 compute Ops. Add a cast to fp16/fp32 before Ops, then add a cast back to fp8.

Note

Must be run after BindTarget, as it relies on target attributes for Functions

Parameters:

promote_dtype – The data type used for type promotion, defaults to float16

Returns:

The pass.