Commit message (Collapse) | Author | Age | Files | Lines | |
---|---|---|---|---|---|
* | Recommend using maturin. (#717) | Laurent Mazare | 2023-09-02 | 1 | -11/+1 |
| | |||||
* | Cleanup the pyo3 setup. (#705) | Laurent Mazare | 2023-09-01 | 1 | -4/+12 |
| | |||||
* | Add a naive conv2d cuda kernel. (#438) | Laurent Mazare | 2023-08-14 | 1 | -0/+2 |
| | | | | | | | | | | | * 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. | ||||
* | Pyo3 dtype (#327) | Laurent Mazare | 2023-08-06 | 1 | -2/+1 |
| | | | | | * Better handling of dtypes in pyo3. * More pyo3 dtype. | ||||
* | Add some tensor creation functions to the pyo3 bindings. (#326) | Laurent Mazare | 2023-08-06 | 1 | -0/+5 |
| | |||||
* | Fix two cuda bugs (matmul and where_cond). | laurent | 2023-06-27 | 1 | -0/+12 |
| | |||||
* | Refactor the hierarchy. | Nicolas Patry | 2023-06-27 | 1 | -12/+0 |
| | |||||
* | Add a makefile for cleaning the kernels code. | laurent | 2023-06-22 | 1 | -0/+12 |