Commit Graph

35 Commits

Author SHA1 Message Date
Laurent Mazare a3f97c143d
Bump the crate version + update CHANGELOG. (#628) 2023-08-27 18:17:11 +01:00
Laurent Mazare e21c686cdc
Fixes for clippy 1.72. (#587) 2023-08-24 17:46:17 +01:00
Laurent Mazare aba1e90797
Add some group parameter to convolutions. (#566)
* Add some group parameter to convolutions.

* Avoid some unnecessary groups checks.

* Move the tensor convolution bits.

* Properh handling of groups.

* Bump the crate version.

* And add a changelog.
2023-08-23 12:58:55 +01:00
Laurent Mazare 9a5c7db91a
Add support for i64 (#563)
* Add the i64 dtype.

* Adapt the cuda kernels.
2023-08-23 10:42:19 +01:00
Laurent Mazare a8f61e66cc
Bump the crates version to 0.1.2. (#522) 2023-08-20 08:07:07 +01:00
Laurent Mazare 531f23b4d0
Rename vec-dot to vec-ops. (#449)
* Rename vec-dot to vec-ops.

* Also bump the crate version.

* Add a currently empty readme.
2023-08-15 10:48:57 +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 Mazare 0b175fcbbd
Fix the pyo3 build for macos. (#324)
* Fix the pyo3 build for macos.

* rustfmt fix.
2023-08-05 14:53:57 +01:00
Laurent Mazare 4fe8a02f88
Update the repo location. (#305) 2023-08-02 11:12:18 +01:00
Laurent Mazare d38943aadc
Add version numbers for all the candle crates (#303)
* Switch to candle-gemm for the time being.

* Add the missing versions.
2023-08-02 10:52:13 +01:00
Laurent Mazare 51e51da896
Rename the candle crate to candle-core (#301)
* Rename to candle-core.

* More candle-core renaming.
2023-08-02 08:20:22 +01:00
Laurent Mazare 104f89df31
Centralize the dependency versions and inherit them. (#177) 2023-07-16 07:47:17 +01:00
Laurent Mazare d1f5d44c04
Reenable pyo3 in the workspace list (#170)
* Enable pyo3 back.

* Adapt the CI.
2023-07-14 19:54:38 +01:00
Nicolas Patry 4ed56d7861 Removing cuda default.
Seems very important for a lot of exploring users usually on laptop
without GPUs.

Adding more README instructions in a follow up.
2023-07-14 16:52:15 +02:00
Laurent Mazare 2bfa791336
Use the same default as pytorch for sum. (#164) 2023-07-13 21:32:32 +01:00
Laurent Mazare 50b0946a2d
Tensor mutability (#154)
* Working towards tensor mutability.

* Use a ref-cell to provide tensor mutability.
2023-07-13 11:04:40 +01:00
Nicolas Patry 868743b8b9 Expanding a bit the README 2023-07-10 12:51:37 +02:00
laurent 5b0ee2e0ba Get cuda to work on pyo3. 2023-07-02 21:04:11 +01:00
laurent fbfe74caab Preliminary pyo3 support for device. 2023-07-02 20:42:55 +01:00
laurent bdb257ceab Add the tensor function. 2023-07-02 20:15:50 +01:00
laurent 78871ffe38 Add dtype support. 2023-07-02 20:12:26 +01:00
laurent d38897461b Add to the example. 2023-07-02 07:37:17 +01:00
laurent 5b8c6764b0 Add matmul/where_cond. 2023-07-02 07:34:14 +01:00
laurent 9a9858bbe0 Expose a couple more ops. 2023-07-02 07:30:00 +01:00
laurent dfe197f791 Handle more input types to create tensors. 2023-07-02 07:19:46 +01:00
laurent 4a28dcf828 Rename the method. 2023-07-02 07:08:11 +01:00
laurent c62cb73a7f Support higher order shapes for conversions. 2023-07-02 07:07:22 +01:00
laurent fa58c7643d Add a trait to avoid repeating the dtype matching. 2023-07-02 06:58:10 +01:00
laurent 2370b1675d More pyo3. 2023-07-01 22:15:58 +01:00
laurent 86df4ad79c Get shape to return a tuple. 2023-07-01 21:34:38 +01:00
laurent fbbde5b02c Add some binary operators. 2023-07-01 21:27:35 +01:00
laurent 42d1a52d01 Add two methods. 2023-07-01 20:55:15 +01:00
laurent 52db2a6849 Apply rustfmt. 2023-07-01 20:37:28 +01:00
laurent ebb0fedf14 Very simple pyo3 bindings for candle. 2023-07-01 20:36:44 +01:00