|
|
53628db3a9
|
Merge pull request #36 from LaurentMazare/fix_example
Simple example fix.
|
2023-06-29 13:36:05 +02:00 |
|
|
|
1913512f42
|
Simple example fix.
|
2023-06-29 11:10:57 +00:00 |
|
|
|
c0719b7781
|
Merge pull request #35 from LaurentMazare/const-scalar
Use broadcasted scalars for const tensors.
|
2023-06-29 12:10:19 +01:00 |
|
|
|
2741b39ad3
|
Use broadcasted scalars for const tensors.
|
2023-06-29 11:56:40 +01:00 |
|
|
|
3872dc4751
|
Merge pull request #19 from LaurentMazare/llama_safetensors
Llama safetensors
|
2023-06-29 12:49:26 +02:00 |
|
|
|
5930168457
|
Merge pull request #34 from LaurentMazare/simpler-dtype-trait
Put more requirements on the withdtype trait.
|
2023-06-29 11:41:17 +01:00 |
|
|
|
b4aab7b95f
|
Put more requirements on the withdtype trait.
|
2023-06-29 11:37:42 +01:00 |
|
|
|
c8fc9da737
|
Merge pull request #33 from LaurentMazare/cuda-map
Simplify the dtype matchings in the cuda backend
|
2023-06-29 10:14:12 +01:00 |
|
|
|
c9c468e1aa
|
Use Map2 for binary ops.
|
2023-06-29 10:09:15 +01:00 |
|
|
|
83c7d660ca
|
Add Map2.
|
2023-06-29 10:05:06 +01:00 |
|
|
|
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 |
|
