tinygrad

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-23 05:48:08 -05:00

Files

geohotstan cea5853cfa add Tensor.scatter (#7737 )

* working I think

* where are my onnx scatter tests??

* forward_only for now

* try if nan hack fix NV

* looks like issue is different... CUDA WHY

* oops that was wrong. Try if this fixes CUDA

* simpler multiply

* actually finish this up tmrw morning :x

* fix tests?

* improve tests

* improve test and implementation

* fix ruff

* complete but lots of expected failure...

* reviewed tests

* add onnx tests

* is this a processing op?

* add return type to indicate that it's not in-place

* final cleanups

* use or and improve tests a little

* add masked_index_select

* call it masked_setitem instead

* try

* FIXED

---------

Co-authored-by: chenyu <chenyu@fastmail.com>

2024-11-27 10:52:04 -05:00

developer

unify to HWQueue [pr] (#7812 )

2024-11-21 10:33:08 +08:00

showcase

Add yolov8 implementation (#806 )

2023-06-16 18:55:19 -07:00

tensor

add Tensor.scatter (#7737 )

2024-11-27 10:52:04 -05:00

abstractions2.py

rename allocator methods to not conflict [pr] (#7788 )