|
|
fd3131a4ce
|
Fix the debug implementation. (#648)
|
2023-08-28 22:51:39 +01:00 |
|
|
|
71518caeee
|
Align tensor device print more with PyTorch (#590)
* Improve tensor print
* Use CudaDevice only if enabled with cuda feature
* run rust fmt
* up
* improve
* rustfmt
|
2023-08-26 11:20:22 +01:00 |
|
|
|
9a5c7db91a
|
Add support for i64 (#563)
* Add the i64 dtype.
* Adapt the cuda kernels.
|
2023-08-23 10:42:19 +01:00 |
|
|
|
674eb35e10
|
Remove some dead-code pragmas. (#137)
|
2023-07-11 09:33:59 +01:00 |
|
|
|
33479c5f1b
|
Add some very simple sum benchmark. (#108)
* Add some very simple sum benchmark.
* Rename the file.
|
2023-07-08 08:39:27 +01:00 |
|
|
|
8ad47907f3
|
Add the kernels.
|
2023-06-30 10:26:56 +01:00 |
|
|
|
b0f5f2d22d
|
Add some display tests + bugfixes.
|
2023-06-27 21:37:28 +01:00 |
|
|
|
8c81a70170
|
PyTorch like display implementation.
|
2023-06-27 21:16:35 +01:00 |
|
|
|
1d504cc6b3
|
Rework the debug trait.
|
2023-06-27 19:10:30 +01:00 |
|
