Francis Lata 3644077a42 [MLPerf][UNet3D] Add DICE loss + metrics (#4204) ... * add DICE loss and metrics * update dice to include reference implementation's link * remove unused imports * remove unnecessary test file and update pred + label for metrics and losses test * add tests to CI + add exclusion of mlperf_unet3d --------- Co-authored-by: chenyu <chenyu@fastmail.com>		2024-04-17 20:09:33 -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	touchup resnet_layer_bench (#4191)	2024-04-16 14:43:00 -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	address kfd feedback (#4087)	2024-04-05 15:24:41 -07: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	rewrite the jit in the context of new schedule (#4162)	2024-04-12 21:54:36 -07: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	use is to compare with enum (#3993)	2024-03-29 13:02:56 -04:00
fuzz_schedule.py	Fuzz all permutations of schedule (#4136)	2024-04-17 05:03:21 +04:00
fuzz_shapetracker.py	minor improvements (#2845)	2023-12-18 22:09:08 -08:00
fuzz_shapetracker_math.py	touchup canonicalize empty mask (#3308)	2024-02-03 21:05:10 -05: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