print_compilation_stage¶
- print_compilation_stage(fn, stage)[source]¶
Print one of the recorded compilation stages for a JIT-compiled function.
The stages are indexed by their Catalyst compilation pipeline name, which are either provided by the user as a compilation option, or predefined in
catalyst.compiler
.Requires
keep_intermediate=True
.- Parameters
fn (QJIT) – a qjit-decorated function
stage (str) – string corresponding with the name of the stage to be printed
Example
@qjit(keep_intermediate=True) def func(x: float): return x debug.print_compilation_stage(func, "HLOLoweringPass")
code/api/catalyst.debug.print_compilation_stage
Download Python script
Download Notebook
View on GitHub