chenyuandGitHub bdcdf1f1a1 jittable masked_select and nonzero (#16170) ...

* jittable masked_select and nonzero

make jittable with `size=`, matches jax

* COMPILE_ONLY

2026-05-12 16:39:36 -04:00

..

codegen

time_call (#16152)

2026-05-12 16:58:28 +03:00

engine

hcq2: remove hcqprogram (#16157)

2026-05-12 18:49:13 +03:00

llm

llm: add markers to --benchmark (#16153)

2026-05-12 20:14:11 +09:00

mixin

fix cumsum dtype for 0d input (#16164)

2026-05-12 14:18:08 -04:00

nn

Revert "onnx: deduplicate simple proto parsers" (#16148)

2026-05-11 21:45:17 -04:00

renderer

Ops.IDIV -> Ops.CDIV, Ops.MOD -> Ops.CMOD (#16093)

2026-05-07 23:18:15 -04:00

runtime

nv: rm _download (#16147)

2026-05-11 19:56:37 -04:00

schedule

Update the spec in spec.py to match the current state (#16132)

2026-05-11 20:07:47 -07:00

uop

allow after on contiguous in spec (#16169)

2026-05-12 13:11:44 -07:00

viz

viz: canonicalize ast for schedule to codegen linking (#16154)

2026-05-12 22:40:21 +09:00

__init__.py

start function and add walk rewrite (#14992)

2026-02-25 13:56:27 +08:00

callify.py

fix: don't copy on slice of DP weight (#16089)

2026-05-07 17:58:01 -07:00

device.py

get_buf() for Buffer (#16134)

2026-05-11 16:36:14 +03:00

dtype.py

cl: image alignment in arch (#16106)

2026-05-08 19:33:33 -04:00

function.py

delete old UOp.size [pr] (#15756)

2026-04-15 23:21:00 -04:00

gradient.py

gradient: add TRUNC backward (#15925)

2026-05-08 16:27:55 -07:00

helpers.py

nv: look for gsp firmware in /lib/firmware (#16068)

2026-05-06 21:35:47 -04:00

py.typed

add a single py.typed (#6083)

2024-08-14 17:31:46 -07:00

tensor.py

jittable masked_select and nonzero (#16170)

2026-05-12 16:39:36 -04:00