Laurent Mazare
21109e1983
Recommend using maturin. ( #717 )
2023-09-02 16:19:35 +01:00
Laurent Mazare
237323c2bc
Cleanup the pyo3 setup. ( #705 )
2023-09-01 14:26:18 +01:00
Laurent Mazare
34f4b3187e
Add a naive conv2d cuda kernel. ( #438 )
...
* Add a naive conv2d cuda kernel.
* Proper conv2d support on the rust side.
* Conv1d testing on gpu.
* Also use the test on gpus.
* Fix the clean-ptx target.
2023-08-14 10:34:42 +01:00
Laurent Mazare
93cfe5642f
Pyo3 dtype ( #327 )
...
* Better handling of dtypes in pyo3.
* More pyo3 dtype.
2023-08-06 10:17:43 +01:00
Laurent Mazare
88bd3b604a
Add some tensor creation functions to the pyo3 bindings. ( #326 )
2023-08-06 06:50:33 +01:00
laurent
380d61e990
Fix two cuda bugs (matmul and where_cond).
2023-06-27 11:31:04 +01:00
Nicolas Patry
d7f729fb8f
Refactor the hierarchy.
2023-06-27 11:57:27 +02:00
laurent
97fe1fac85
Add a makefile for cleaning the kernels code.
2023-06-22 13:57:51 +01:00