Guilherme Leobas

Removes `context.compile_internal` where easy

16

Removes context.compile_internal where easy * np.ravel * array.flatten * array.sort * round * not in * ~math.gcd~ * np.argsort

Caching is useless on higher order functions

2

This is more of a feature request/discussion than a bug. ### Reproducer ```python from numba import njit @njit def incr(x): return x + 1 @njit(cache=True) def foo(f, x): return f(x)...

Add axis support to np.take

2

Non-scientific script to measure time: ```python import time from numba import njit import numpy as np @njit def foo(a, indices, axis): return np.take(a, indices, axis) def measure_time(fn, *args): start =...

Replace `context.compile_internal` by `@overload` in cmathimpl.py

12

~This is more of an exploratory work, and it is not ready yet.~ ~The idea is to replace all occurrences of `compile_internal` by `call_overload`, introduced in this PR. Some of...

Introduce `context.call_overload` and raise dynamic exception in `guard_match_core_dims`

2

PR is separated into four commits: * 02d0d45: Add `overload_method(UniTuple, '__str__')` and tests * 0be554f: Add `context.call_overload` * 2c4324e: Update `guard_match_core_dims` to call an overload that can emit a dynamic...

add ufunc.reduceat support

2

This PR adds support for `ufunc.reduceat` method inside `@jit` decorator. Needs the following PRs to be merged first: * https://github.com/numba/numba/pull/9297 * https://github.com/numba/numba/pull/9296 * https://github.com/numba/numba/pull/9295

`@torch.compile` some tutorials

6

## Description This PR attempts to compile three tutorials: * neural_tangent_kernels * ensembling * per_sample_grads To compile with `fullgraph=True`, one needs pytorch with the changes from https://github.com/pytorch/pytorch/pull/129091. ## Performance gain...

### Reproducer ```python from numba import int64, njit from numba.experimental import jitclass spec = [ ('value', int64), ] @jitclass(spec) class Bag(object): def __init__(self, value): self.value = value @njit(cache=True) def foo(a):...

Raise exception if vmap (eager) calls compiled function

3

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #140439 Fixes #138422 This is not a proper fix for #140439, but more of a way to prevent a user from seeing...

Set `enable_trace_contextlib_contextmanager` flag to True

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #141055 * #138203 * __->__ #140604 * #136033 * #141021 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng...