issues
search
hikettei
/
Caten
[wip] Deep Learning Compiler based on Polyhedral Compiler and Light-weight IRs based on Optimizing Pattern Matcher
https://hikettei.github.io/Caten/
Other
15
stars
1
forks
source link
Fix: !randint dtype inference
#129
Open
hikettei
opened
3 weeks ago
hikettei
commented
3 weeks ago
[ ] Add dtype check here (
https://github.com/hikettei/Caten/blob/main/source/apis/module.lisp#L22
)
[ ] Add
(!cast out :dtype (dtype-of x))
here
https://github.com/hikettei/Caten/blob/main/source/apis/initializers.lisp#L73
[ ] Adding tests more decent tests for !randint, !randn
(!cast out :dtype (dtype-of x))
here https://github.com/hikettei/Caten/blob/main/source/apis/initializers.lisp#L73