chenyu 7816c3b692 onnx update for trilu and argmax (#3283) ...

* support 0 in shape for tril and triu

* select_last_index for ArgMax and ArgMin

* pass **kwargs

2024-01-30 18:39:16 -05:00

..

external_benchmark_hip_compile.py

hip compile speed (#2606)

2023-12-04 13:47:40 -08:00

external_benchmark_load_stable_diffusion.py

gate METAL_FAST_LOAD

2023-12-01 15:28:40 -08:00

external_cl_half_max.py

use default dict for external_model_benchmark (#2592)

2023-12-03 15:25:43 -08:00

external_jit_failure.py

fix jit realize issue (#3258)

2024-01-26 18:27:35 -08:00

external_llama_eval.py

ruff checks the max line length is 150 (#2734)

2023-12-12 17:34:47 -08:00

external_model_benchmark.py

separate try except blocks in onnx2torch in model benchmark (#3126)

2024-01-15 00:39:33 -05:00

external_multi_gpu.py

s/lazydata.realized/lazydata.base.realized/g (#2914)

2023-12-22 14:45:13 -05:00

external_osx_profiling.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

external_test_dist_collectives.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

external_test_dist_world.py

more test cleanups (#2631)

2023-12-05 16:17:57 -08:00

external_test_embedding.py

make embedding and GPT-2 fast (#1631)

2023-08-22 15:14:38 -07:00

external_test_example.py

Device._buffers -> Device._devices (#3052)

2024-01-08 21:30:38 -05:00

external_test_image.py

fixes (#1893)

2023-09-22 07:20:27 +08:00

external_test_jit_on_models.py

move dtypes to dtype.py (#2964)

2024-01-01 14:58:48 -08:00

external_test_onnx_backend.py

onnx update for trilu and argmax (#3283)

2024-01-30 18:39:16 -05:00

external_test_opt.py

move globalcounters to ops (#2960)

2024-01-01 14:21:02 -08:00

external_test_optim.py

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

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

external_test_speed_llama.py

cached size (#3060)

2024-01-09 16:37:37 -08:00

external_test_uops_graphing.py

cleanups to save lines and files (#2577)

2023-12-02 16:29:56 -08:00

external_test_whisper_librispeech.py

whisper: support audio >30s (#2378)

2023-11-21 14:37:51 -08:00

external_test_yolo.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

external_test_yolov8.py

ruff checks the max line length is 150 (#2734)

2023-12-12 17:34:47 -08:00

fuzz_linearizer.py

fuzzer validates outputs + (partially) oob accesses (#3178)

2024-01-19 13:34:51 -05:00

fuzz_shapetracker_math.py

add canonicalization to View.create (#3280)

2024-01-30 10:26:48 -08:00

fuzz_shapetracker.py

minor improvements (#2845)

2023-12-18 22:09:08 -08:00

fuzz_symbolic.py

Variable.num -> NumNode (#2354)

2023-11-18 15:45:52 -05:00

graph_batchnorm.py

with Tensor.train() (#1935)

2023-09-28 18:02:31 -07:00