3 people fc11808a79 initialize Tensor grad same type as self (#3613) ...

* initialize Tensor grad same type as self

* also test different default float

* check dtype + try/finally

* don't test_gradient_dtype if f16 is not supported

* fix bad merge

---------

Co-authored-by: chenyu <[email protected]>

2024-03-22 20:33:18 -04:00

..

codegen

replace raise Exception with specific errors (#3874)

2024-03-22 12:32:21 -04:00

features

debug: add optional detailed BEAM_LOG logging (#3883)

2024-03-22 19:23:31 -04:00

nn

fix assign (#3878)

2024-03-22 11:48:48 -07:00

renderer

hotfix: hip bfloat formatting

2024-03-22 11:52:05 -07:00

runtime

init multidevice cuda graph (#3858)

2024-03-22 13:49:48 -07:00

shape

fix View.pad arg type (#3845)

2024-03-20 19:36:02 -04:00

__init__.py

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

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

device.py

log optimized kernels and a script to compare with non-optimized ones (#3829)

2024-03-20 19:22:08 -04:00

dtype.py

Rename .sz to .count on DType (#3413)

2024-02-15 15:03:49 +01:00

helpers.py

Ring allreduce try 2 (#3852)

2024-03-21 19:17:51 -04:00

lazy.py

Ring allreduce try 2 (#3852)

2024-03-21 19:17:51 -04:00

mlops.py

reciprocal mlop (#3694)

2024-03-11 16:08:46 -07:00

ops.py

simple LoadOps.ASSIGN (#3745)

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

realize.py

fix assign (#3878)

2024-03-22 11:48:48 -07:00

tensor.py

initialize Tensor grad same type as self (#3613)

2024-03-22 20:33:18 -04:00