Commit Graph

469 Commits

Author SHA1 Message Date
367170da45 Also use Map1 for embedding. 2023-06-29 09:45:27 +01:00
8ad03a5fb6 Use Map1 on unary ops. 2023-06-29 09:37:38 +01:00
fff13dbb4e Factorize the kernel naming scheme. 2023-06-29 09:29:59 +01:00
d3c7b0d168 Use Map1 for sum. 2023-06-29 09:27:07 +01:00
122e334d0c Simplify the pattern matching logic in the cuda backend. 2023-06-29 09:21:11 +01:00
eda46d2df2 Merge pull request #32 from LaurentMazare/running_less_ci
Running CI only when pushing on `main` and on pull request.
2023-06-29 08:24:31 +02:00
5f65d46c32 Merge pull request #29 from LaurentMazare/cpu-map
Switch from a macro to a trait to make things more generic.
2023-06-29 05:27:59 +01:00
f08f146348 Merge pull request #31 from LaurentMazare/fix_hub
Fixing hub test.
2023-06-29 00:22:17 +02:00
0862e7d9e9 Windows 2019 is slower to load (I guess less availability). 2023-06-28 22:21:38 +00:00
d3000ac9eb Running CI only when pushing on main and on pull request. 2023-06-28 22:20:31 +00:00
beccf673f4 Fixing hub test. 2023-06-28 22:16:46 +00:00
eaa3ce359e Cosmetic change. 2023-06-28 22:02:23 +01:00
1328b5cb20 Factor some code out. 2023-06-28 21:56:44 +01:00
c583ee0f2c Add map2. 2023-06-28 21:38:01 +01:00
46c07b924c Tweak some comment. 2023-06-28 21:10:54 +01:00
2ae368e98e Switch from a macro to a trait to make things more generic. 2023-06-28 21:06:56 +01:00
0cfa21f26a Merge pull request #27 from LaurentMazare/layout-refactor
Refactor the stride/shape handling
2023-06-28 15:59:53 +01:00
ece3ec6167 Final updates -> moving to deterministic for easier comparison. 2023-06-28 14:56:39 +00:00
926fffa0b7 Ok. 2023-06-28 14:56:39 +00:00
e29dae044d Tmp. 2023-06-28 14:56:38 +00:00
6c9e6b5a99 Get the cuda tests to pass. 2023-06-28 15:53:23 +01:00
8b4b2d1830 Merge pull request #28 from LaurentMazare/fix_hub
Remove the unecessary file lock, attempt to rename before copying.
2023-06-28 16:49:07 +02:00
3f0d9fbb25 Adapt the cuda bits. 2023-06-28 15:43:03 +01:00
cfdfc04d5c Remove the unecessary file lock, attempt to rename before copying. 2023-06-28 16:42:26 +02:00
cca699be6c Fix some cpu issue. 2023-06-28 15:09:15 +01:00
1c755c0e5b Remove some todos. 2023-06-28 14:33:06 +01:00
caafef6cc1 Get the cpu tests to run. 2023-06-28 14:32:02 +01:00
14449ff80c Get the cpu backend to compile. 2023-06-28 14:12:38 +01:00
54a6c40f27 Propagate the changes on the cpu backend. 2023-06-28 14:00:49 +01:00
303b853098 Propagate the layout refactoring. 2023-06-28 13:42:23 +01:00
30b355ccd2 Simplify the narrow implementation. 2023-06-28 13:09:59 +01:00
c1bbbf94f6 Start refactoring the stride. 2023-06-28 12:57:30 +01:00
d461d9d751 Merge pull request #26 from LaurentMazare/narrow-grad
Add the grad for narrow.
2023-06-28 11:46:13 +01:00
666d6dbcac Merge remote-tracking branch 'origin/main' into narrow-grad 2023-06-28 11:45:46 +01:00
2998ff6ef7 Merge pull request #25 from LaurentMazare/fix_hub
Fix flaky test ?
2023-06-28 12:42:38 +02:00
7938d2b848 Add the grad for narrow. 2023-06-28 10:46:00 +01:00
9c86e4afa8 Fix flaky test ? 2023-06-28 11:40:41 +02:00
d0ff3b2d13 Merge pull request #24 from LaurentMazare/more-grads
Support gradients for reshape and where_cond.
2023-06-28 10:04:51 +01:00
615196e7be Add more gradients. 2023-06-28 09:59:52 +01:00
50eff0005b Merge pull request #23 from LaurentMazare/relu
Add the relu op.
2023-06-28 09:44:24 +01:00
1ce3843cab Add the relu op. 2023-06-28 09:38:54 +01:00
b805c4114b Merge pull request #22 from LaurentMazare/more-cuda-testing2
Again more cuda testing.
2023-06-28 09:01:25 +01:00
19183b8e4f Factor out the gemm bits. 2023-06-28 08:51:13 +01:00
0417d9cec8 Add more cuda testing again. 2023-06-28 08:33:43 +01:00
64c6bc4f5e Merge pull request #21 from LaurentMazare/more-cuda-tests
Also run the backprop tests on cuda.
2023-06-28 08:19:01 +01:00
395c84e80a Also run the backprop tests on cuda. 2023-06-28 08:15:03 +01:00
a457020d50 Merge pull request #20 from LaurentMazare/tensor-display
Add some pretty print display to Tensors
2023-06-27 21:53:09 +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
934655a60d Add squeeze/unsqueeze/stack. 2023-06-27 19:32:00 +01:00