3 people 755dfb243b rename CPU_COUNT to NUM_CPU_THREADS with cgroup awareness (#17263) ...

Rename CPU_COUNT to NUM_CPU_THREADS so it can be overridden via env var.
Default uses _get_cpu_count() which respects cgroup limits:
  - os.process_cpu_count() on Python 3.13+
  - /sys/fs/cgroup/cpu.max on cgroup v2
  - /sys/fs/cgroup/cpu/cpu.cfs_quota_us on cgroup v1
  - os.sched_getaffinity(0) fallback

Use NUM_CPU_THREADS.value in the dataloader instead of cpu_count(),
and update export_model.py and all renderer references.

Co-authored-by: teeny-runner <runner@teeny>

2026-07-28 15:33:57 -07:00

..

codegen

switch _device_num to AxisType.DEVICE range (kimi) (#17252)

2026-07-28 13:52:47 -07:00

engine

switch _device_num to AxisType.DEVICE range (kimi) (#17252)

2026-07-28 13:52:47 -07:00

llm

remove unneeded expand [PR] (#17176)

2026-07-24 10:03:46 -04:00

mixin

delete explicit casts [pr] (#17255)

2026-07-28 13:57:13 -04:00

nn

switch _device_num to AxisType.DEVICE range (kimi) (#17252)

2026-07-28 13:52:47 -07:00

renderer

rename CPU_COUNT to NUM_CPU_THREADS with cgroup awareness (#17263)

2026-07-28 15:33:57 -07:00

runtime

remove unneeded default args in renderers (#17262)

2026-07-28 14:53:04 -07:00

schedule

switch _device_num to AxisType.DEVICE range (kimi) (#17252)

2026-07-28 13:52:47 -07:00

uop

switch _device_num to AxisType.DEVICE range (kimi) (#17252)

2026-07-28 13:52:47 -07:00

viz

polish the viz readme (#17236)

2026-07-27 22:03:13 +09:00

__init__.py

start function and add walk rewrite (#14992)

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

callify.py

UOp.const(dtype=None) infers from from_py [pr] (#17253)

2026-07-28 12:06:55 -04:00

device.py

Revert "64-bit UOp.variable support (#17246)" (#17248)

2026-07-28 00:00:26 -04:00

dtype.py

flip from_py to use weak dtypes [pr] (#17229)

2026-07-27 10:07:00 -04:00

function.py

Buffer arg -> max_numel [PR] (#16752)

2026-06-25 14:50:20 -04:00

helpers.py

rename CPU_COUNT to NUM_CPU_THREADS with cgroup awareness (#17263)

2026-07-28 15:33:57 -07:00

py.typed

add a single py.typed (#6083)

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

tensor.py

UOp.const(dtype=None) infers from from_py [pr] (#17253)

2026-07-28 12:06:55 -04:00