George Hotz 2ae4f45272 WIP PM4 Support (#4110) ...

* pm4 kernel launch works

* disable USE_THREAD_DIMENSIONS

* add kernel code

* work on real pm4

* pm4 signal

* same

* gate pm4

* hcq tests pass

* ops passes

* pm4 is closer

* pm4 debug (#4165)

* start debug tests passing

* prg

* smth

* hdp flush

* cleaner 1

* do not need this

* logs not need

* small things

* linter

* remove AQL

* test hcq

* fix tests

* it's subtracting, it shouldn't be -1

* pm4 changes (#4251)

* not need this anymore

* sdma signal with non atomic

---------

Co-authored-by: nimlgen <138685161+nimlgen@users.noreply.github.com>

2024-04-23 08:31:27 +04:00

..

mlperf_resnet

PolynomialDecayWithWarmup + tests (#3649)

2024-03-07 18:53:36 -05:00

mlperf_unet3d

[MLPerf][UNet3D] Add DICE loss + metrics (#4204)

2024-04-17 20:09:33 -04: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_benchmark_multitensor_allreduce.py

move GlobalCounter to helpers (#4002)

2024-03-30 00:30:30 -04:00

external_benchmark_resnet.py

more realistic edge behavior for resnet benchmark (#4231)

2024-04-19 20:07:46 -04:00

external_cl_half_max.py

use default dict for external_model_benchmark (#2592)

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

external_hip_compiler_bug.py

CompiledASTRunner -> CompiledRunner (#4148)

2024-04-11 08:49:52 -07: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

Add driving monitoring model to benchmarks (#4134)

2024-04-10 14:27:03 -04: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_embedding.py

make embedding and GPT-2 fast (#1631)

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

external_test_example.py

numpy device + pickle it (#4120)

2024-04-09 13:19:30 -07:00

external_test_hcq.py

WIP PM4 Support (#4110)

2024-04-23 08:31:27 +04:00

external_test_hip_compile.py

split to schedule.py (#3949)

2024-03-26 21:02:46 -07:00

external_test_hsa_driver.py

JitItem -> ExecItem (#4146)

2024-04-11 08:24:57 -07:00

external_test_image.py

fixes (#1893)

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

external_test_jit_on_models.py

create engine folder and move code (#3948)

2024-03-26 20:38:03 -07:00

external_test_losses.py

[MLPerf][UNet3D] Add DICE loss + metrics (#4204)

2024-04-17 20:09:33 -04:00

external_test_mamba.py

external that test

2024-03-29 19:35:50 -07:00

external_test_metrics.py

[MLPerf][UNet3D] Add DICE loss + metrics (#4204)

2024-04-17 20:09:33 -04:00

external_test_mnist_data_select.py

add quick external data select test

2024-03-02 05:38:32 -08:00

external_test_onnx_backend.py

update onnx to 1.16.0 (#4127)

2024-04-10 11:19:13 -04:00

external_test_opt.py

cleanup external_test_opt (#4234)

2024-04-20 04:00:08 -04:00

external_test_optim.py

add lars to nn (#3750)

2024-03-24 11:43:12 -04:00

external_test_speed_llama.py

multitensor shouldn't recompile (#4164)

2024-04-13 00:03:48 -07:00

external_test_uops_graphing.py

split to schedule.py (#3949)

2024-03-26 21:02:46 -07: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_kfd.py

more work on kfd (#4079)

2024-04-05 08:36:36 -07:00

fuzz_linearizer.py

clean up update stats (#4226)

2024-04-19 15:41:30 +04:00

fuzz_schedule.py

fuzz schedule context vars (#4223)

2024-04-19 13:16:25 +03:00

fuzz_shapetracker_math.py

touchup canonicalize empty mask (#3308)

2024-02-03 21:05:10 -05: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

speed_beam_v_hcopt.py

LinearizerOptions -> CompilerOptions (#3978)

2024-03-28 17:50:23 -04:00

speed_compare_cuda_ptx.py

CompiledASTRunner -> CompiledRunner (#4148)

2024-04-11 08:49:52 -07:00

verify_kernel.py

test: add fuzz_matmul and better debugging for simple_matmul (#4199)

2024-04-16 23:40:31 -04:00