qazalandGitHub 0d4887e9df use UOps.WMMA everywhere (#6255) ...

* add UOps.WMMA_AXIS

* delete ReduceOps.WMMA from ops

2024-08-23 15:03:26 -04:00

..

codegen

use UOps.WMMA everywhere (#6255)

2024-08-23 15:03:26 -04:00

engine

start graph rewrite in the scheduler (#6248)

2024-08-23 13:15:55 +03:00

nn

Tensor.prod (#6250)

2024-08-23 10:06:32 -04:00

renderer

remove UnaryOps.NEG (#6238)

2024-08-22 14:21:39 -04:00

runtime

nv fix size in SET_SEMAPHORE_A (#6213)

2024-08-21 01:47:10 +03:00

shape

re:6125 switch real_size to use uops [run_process_replay] (#6138)

2024-08-19 13:20:24 -04:00

__init__.py

add mnist tutorial

2024-04-25 16:08:32 +08:00

device.py

RUF018 assignment-in-assert [run_process_replay] (#6172)

2024-08-19 00:34:52 -04:00

dtype.py

threefry touchup [run_process_replay] (#6169)

2024-08-18 23:01:24 -04:00

function.py

Tensor.prod (#6250)

2024-08-23 10:06:32 -04:00

helpers.py

Support gunzip in fetch (#6176)

2024-08-19 12:04:40 -07:00

lazy.py

Tensor.prod (#6250)

2024-08-23 10:06:32 -04:00

multi.py

MLB support reshape for uneven shards (#5804)

2024-08-01 02:36:03 -07:00

ops.py

use UOps.WMMA everywhere (#6255)

2024-08-23 15:03:26 -04:00

py.typed

add a single py.typed (#6083)

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

tensor.py

Tensor.prod (#6250)

2024-08-23 10:06:32 -04:00