wozeparrot 9a9cac58f9 add lars to nn (#3750) ...

* feat: add lars

* feat: don't remove this comment

* clean: smaller diff

* clean: shorter line

* feat: remove mlperf lars, switch resnet

* fix: fully remove mlperf lars

* clean: comment

* feat: contiguous

* feat: no weight decay on skip params

* feat: optimizergroup

* feat: classic momentum

* fix: pylint

* clean: move comment

* fix: correct algo

* feat: lrschedulergroup

* feat: skip list tests

* feat: :| forgot that params are a thing

* feat: remove skip_list params from main params

* feat: set moment

---------

Co-authored-by: chenyu <chenyu@fastmail.com>

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

..

accel

move things, clean up extra (#2292)

2023-11-13 20:18:40 -08:00

assembly

move dtypes to dtype.py (#2964)

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

backends

remove hip backend (#3783)

2024-03-17 10:12:16 -07:00

datasets

add a better error message for resnet training (#3836)

2024-03-20 09:22:15 -07:00

gemm

search: change to use "spawn" and limit the number of tasks per child (#3862)

2024-03-21 21:23:36 -07:00

hip_gpu_driver

nv ioctl sniffer (#3892)

2024-03-23 00:29:30 -07:00

hiprtc

use comgr to compile (#3248)

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

junk

coder.py can write and run code (#2439)

2023-11-25 12:27:54 -08:00

models

restrict assignment to base (#3809)

2024-03-18 15:33:06 -07:00

nv_gpu_driver

nv ioctl sniffer (#3892)

2024-03-23 00:29:30 -07:00

optimization

fuzz_linearizer: add --ast and --file params to read kernels (#3877)

2024-03-22 14:27:40 -04:00

qcom_gpu_driver

start Qualcomm GPU driver (#2804)

2023-12-16 23:10:50 -08:00

archprobe.py

move dtypes to dtype.py (#2964)

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

augment.py

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

autopad.py

move create schedule and delete old API (#3377)

2024-02-12 18:10:45 +01:00

disk_read_speed.py

fast hip read (#3014)

2024-01-05 10:33:13 -08:00

dump_cache.py

wow how did i think that was okay (#2339)

2023-11-16 21:21:11 -08:00

export_model.py

move graph.py and jit.py into features (#3376)

2024-02-12 17:34:34 +01:00

gradcheck.py

Fix: Jacobian tests [WIP] (#1126)

2023-07-05 15:36:22 -07:00

hip_events.py

move autogen to runtime/autogen (#3254)

2024-01-26 12:44:19 -08:00

introspection.py

move globalcounters to ops (#2960)

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

lr_scheduler.py

add lars to nn (#3750)

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

multitensor.py

multitensor start (#2676)

2023-12-07 17:07:05 -08:00

onnx_ops.py

simple LoadOps.ASSIGN (#3745)

2024-03-14 20:44:34 -07:00

onnx.py

Fix: Always cast ONNX Slice op arguments into ints (#3317)

2024-02-04 18:40:48 -05:00

ring_copy.py

ring copy example (#3185)

2024-01-19 23:34:30 -05:00

thneed.py

new style device (#2530)

2023-11-30 17:07:16 -08:00

to_movement_ops.py

fixup to_movement_ops and add back to CI (#3881)

2024-03-22 18:14:49 -04:00

training.py

move graph.py and jit.py into features (#3376)

2024-02-12 17:34:34 +01:00

transfer_speed.py

hotfix: copy size is in bytes

2024-01-17 16:44:15 +00:00