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