wozeparrot 29d5801387 distributed collectives (#1519) ...

* feat: world

* feat: tests

* feat: no more backwards

* feat: recv into

* feat: whoops

* feat: test in ci

* feat: some debug logging

* feat: workflow naming

* feat: need to set pythonpath

* feat: just send to same device

* feat: allreduce

* feat: test

* feat: need contiguous

* feat: test in ci

* feat: exit with correct code

* feat: don't need that

* feat: opencl wait_for just doesn't work

* feat: synchronize on out

* feat: try?

* feat: try again?

* feat: add extra realizes

* feat: print

* feat: seed

* feat: tol

* feat: test ones and zeros

* feat: remove print

* feat: are you just flaky

* feat: seperate scatter and gather?

* feat: just try synchronizing

* feat: remove print again

* feat: bring back difference

* feat: no sync

* feat: revert that

* feat: back to wait_for

* fix: typo

2023-08-11 10:22:07 -07:00

..

external

distributed collectives (#1519)

2023-08-11 10:22:07 -07:00

extra

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00

models

Raise memory limit for CIFAR test (#1499)

2023-08-08 19:40:56 -04:00

unit

move renamer to linearizer (#1442)

2023-08-05 08:53:25 -07:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

Docker fix (#1039)

2023-06-25 10:38:58 -07:00

test_assign.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_conv_shapetracker.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_conv.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_custom_function.py

flake8 (#1323)

2023-07-24 11:19:58 -04:00

test_dtype.py

llvmir support for bool <-> float casting (#1492)

2023-08-09 13:12:52 -04:00

test_gc.py

fix up eye and fix gc test

2023-02-27 06:53:18 -08:00

test_helpers.py

fix: Context (#1430)

2023-08-04 10:53:48 -04:00

test_jit.py

use allclose instead of equals in test_jit (#1504)

2023-08-08 22:22:17 -07:00

test_lazybuffer.py

just cmplt (#1493)

2023-08-08 13:58:10 -07:00

test_linearizer.py

dedup kernel args v2 (#1272)

2023-07-18 20:03:42 -07:00

test_net_speed.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_nn.py

Limit dims based on max size (#1390)

2023-07-31 19:18:19 -07:00

test_ops.py

cleanup mlops (#1521)

2023-08-10 19:53:28 -07:00

test_optim.py

Add WEBGPU tests to CI (#1463)

2023-08-06 10:32:01 -07:00

test_randomness.py

rng hlops: add normal and kaiming_normal (#1378)

2023-07-31 10:37:02 -07:00

test_specific_conv.py

CI < 5 minutes (#1252)

2023-07-23 13:00:56 -07:00

test_speed_v_torch.py

Add WEBGPU tests to CI (#1463)

2023-08-06 10:32:01 -07:00

test_symbolic_shapetracker.py

Support constant expand to symbolic shape (#1411)

2023-08-02 21:21:22 -07:00

test_tensor.py

Add WEBGPU tests to CI (#1463)

2023-08-06 10:32:01 -07:00

test_uops.py

cleanup mlops (#1521)

2023-08-10 19:53:28 -07:00

test_webgpu.js

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00