chenyuandGitHub f363f39e83 fix dtype of const folded sum (#4349) ...

const folding sum should return in the same dtype the same as regular sum, which can be different from input dtype

2024-04-29 11:40:45 -04:00

..

codegen

hotfix: remove double assignment (#4340)

2024-04-28 13:41:31 -04:00

engine

save schedule graph pre validation (#4317)

2024-04-27 12:06:15 +03:00

features

fix dtype of const folded sum (#4349)

2024-04-29 11:40:45 -04:00

nn

use at least float32 for optim.lr (#4297)

2024-04-25 14:42:28 -04:00

renderer

Ptx beam fix (#4296)

2024-04-25 15:39:39 -04:00

runtime

raise error when too many resources requested in nv (#4324)

2024-04-27 23:48:51 +03:00

shape

Fix in _reshape_mask (#4332)

2024-04-28 11:57:39 -04:00

__init__.py

add mnist tutorial

2024-04-25 16:08:32 +08:00

buffer.py

new memory scheduler with explicit refcounts (#4198)

2024-04-17 08:46:47 +04:00

device.py

more docs (#4271)

2024-04-24 10:52:42 +08:00

dtype.py

rename Scalar to ConstType and cast_scalar to as_const (#3946)

2024-03-26 22:39:58 -04:00

function.py

fix mean of half tensor if sum is greater than hlaf.max (#4327)

2024-04-28 18:04:54 -04:00

helpers.py

search: add additional 7 factors to the action space (#4256)

2024-04-22 19:14:23 -04:00

lazy.py

fix dtype of const folded sum (#4349)

2024-04-29 11:40:45 -04:00

ops.py

si.inputs+outputs -> bufs (#4279)

2024-04-24 15:12:34 +08:00

tensor.py

use tolist instead of numpy for extracting parameters in onnx (#4333)

2024-04-29 10:48:20 -04:00