I’m wondering if there are places to register callbacks before and after compilation? I’d like to “pause” timing tests during jit compilation, as compilation time is not relevant. (However, I don’t want to go to eager compilation everywhere.)

Hi @shaunc No guarantee, but I think you can use the event API as follows: from time import perf_counter from dataclasses import dataclass from numba.core import event from numba.core.registry import CPUDispatcher @dataclass class Timing: tic: float | None = None toc: float | None = Non…

Hooks for compilation start/end

Numba Community Support

nelson2005 September 4, 2023, 9:57pm 3

I’m not familiar with the hooks, but I get the total compilation time for a function by summing the pipeline steps, inspired by the @stuartarchibald here

Topic		Replies	Views
Reduce Compilation Time Support: How do I do ...?	8	1369	November 3, 2021
Compilation pipeline, compile time and vectorization Numba	8	1117	January 14, 2023
Tips or tricks for speeding up compilation time on first call of large Numba-jitted NumPy-containing functions? Community Support	3	570	July 28, 2020
Why am i getting different performance speeds for the "same" decorator? Community Support	11	781	March 4, 2021
Lets add standalone compilation like AssemblyScript Development	1	814	June 30, 2020

Hooks for compilation start/end

Related Topics