Akshay Kashyap 018bd29e37 Enable Multi-Output Export (#2179) ...

* Enable Multi-Output Export

* Add test

* Update examples and lint

* fix padding

* test ops

* dummy commit to rerun test

* revert cuda lint

* Enforce tuple/list of tensors

* subscripted generics

* put back webgpu test

* Re-enable WebGPU Efficientnet test

2023-10-30 18:42:26 -07:00

..

external

perf: lazyop as dataclass (#1603)

2023-10-25 17:54:30 -04:00

extra

Enable Multi-Output Export (#2179)

2023-10-30 18:42:26 -07:00

models

perf: lazyop as dataclass (#1603)

2023-10-25 17:54:30 -04:00

unit

use unique table name for each disk_cache test (#2184)

2023-10-30 13:49:49 -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

helpers.py

perf: lazyop as dataclass (#1603)

2023-10-25 17:54:30 -04:00

test_allocators.py

no import when Python is shutting down (#1875)

2023-09-20 12:47:02 -04:00

test_assign.py

corealize + remove realize from lazybuffer (#1968)

2023-10-04 10:59:31 -07:00

test_cache_collector.py

fix jitted dist (#1955)

2023-10-02 11:45:13 -04:00

test_conv_shapetracker.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

test_conv.py

fix wino mask upcast calculation (#2057)

2023-10-18 16:54:48 -07:00

test_custom_function.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_dtype.py

Fix llvm int->bool cast (#2164)

2023-10-30 15:28:23 -07:00

test_exec.py

Refactor hipgraph (#2141)

2023-10-24 15:45:56 -04:00

test_gc.py

fix up eye and fix gc test

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

test_jit.py

unrealized consts everywhere (#1963)

2023-10-04 01:48:10 -07:00

test_kernel_cache.py

kernel cache (#2035)

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

test_lazybuffer.py

corealize + remove realize from lazybuffer (#1968)

2023-10-04 10:59:31 -07:00

test_lazyop.py

Q network for pruning BEAM / uops deduping / BEAM_ESTIMATE (#2142)

2023-10-27 10:53:06 -10:00

test_linearizer_failures.py

start work on auto opt (#2034)

2023-10-11 12:54:53 -07:00

test_linearizer.py

Revert "optimizer: simplify GROUP and LOCAL to have one of each (#2162)" (#2182)

2023-10-30 10:22:26 -07:00

test_net_speed.py

CI < 5 minutes (#1252)

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

test_nn.py

fix wino mask upcast calculation (#2057)

2023-10-18 16:54:48 -07:00

test_ops.py

split test_conv2d

2023-10-18 14:00:50 -07:00

test_optim.py

use class Foo: instead of class Foo(): (#1797)

2023-09-06 12:20:25 -07:00

test_randomness.py

Tensor.uniform set default to standard uniform (#2158)

2023-10-27 16:15:30 -04:00

test_schedule.py

add mops to graph, refactor IMAGE (#2100)

2023-10-17 21:27:51 -07:00

test_search.py

add test coverage for search (#2104)

2023-10-19 17:06:47 -07:00

test_specific_conv.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

test_speed_v_torch.py

wmma: add test and tensor core shape (#1925)

2023-09-28 18:04:28 -07:00

test_symbolic_jit.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_symbolic_ops.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_symbolic_shapetracker.py

Variable.bind newer (#2017)

2023-10-10 10:03:01 -07:00

test_tensor.py

added from unaligned np test (#2134)

2023-10-23 11:38:57 -04:00

test_uops.py

print test durations and add speed (#2107)

2023-10-18 13:46:42 -07:00

test_webgpu.js

Enable Multi-Output Export (#2179)

2023-10-30 18:42:26 -07:00

test_winograd.py

print test durations and add speed (#2107)

2023-10-18 13:46:42 -07:00