dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
488 Commits 49 Branches 17 Tags 90 MiB
Commit Graph

67 Commits

Author SHA1 Message Date
Nathaniel Simard c74e75f748
Fix/wgpu/max pool2d backward (#613) 2023-08-09 16:45:49 -04:00
Caio Piccirillo 1d3bbaab13
Typos (#608) 2023-08-08 17:57:51 -04:00
Nathaniel Simard 441a7011ce
Feat/tensor casting (#604) 2023-08-08 10:02:17 -04:00
Nathaniel Simard 8bc687e1bb
WGPU use best limits for the adaptor (#601) 2023-08-07 15:10:21 -04:00
Gadersd ed255c5561
Use buffered io for massive performance gains when loading and saving… (#593) 2023-08-06 12:56:27 -04:00
Nathaniel Simard 8436d4ff66
Feat/tensor/adaptive avg pool2d (#572) 2023-08-04 10:23:59 -04:00
dependabot[bot] b7ad23bd87
Update serial_test requirement from 0.5.0 to 2.0.0 (#579) 
Updates the requirements on [serial_test](https://github.com/palfrey/serial_test) to permit the latest version.
- [Release notes](https://github.com/palfrey/serial_test/releases)
- [Commits](https://github.com/palfrey/serial_test/compare/v0.5.0...v2.0.0)

---
updated-dependencies:
- dependency-name: serial_test
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2023-08-03 09:21:47 -04:00
Louis Fortier-Dubois d5f9f69cea
Refactor/wgpu/prng (#576) 2023-08-02 16:08:50 -04:00
mmalczak 73fb0eaa7e
Addition of abs tensor opperator #506 (#553) 2023-08-01 18:25:14 -04:00
Louis Fortier-Dubois 87125da6c9
Feat/wgpu/prng bernoulli (#571) 2023-08-01 12:54:22 -04:00
Louis Fortier-Dubois a69788ad4b
Feat/wgpu/prng normal (#570) 2023-08-01 10:59:49 -04:00
Louis Fortier-Dubois daedec6f6d
Feat/wgpu/prng (#560) 
* wip

* wip

* default prng working but not perfectly random

* format

* format

* fix prng algo

* clippy

* uniform prng

* refactor tests for prng

* cleanup
 2023-08-01 09:26:49 -04:00
Louis Fortier-Dubois aa4af29e3f
Matmul speedup (contiguous load) (#559) 2023-07-28 10:41:27 -04:00
Dilshod Tadjibaev 74c41bdda2
Add clamp, clamp_min, clamp_max tensor ops (#550) 2023-07-26 20:02:38 -04:00
Nathaniel Simard 0a5a2d729a
chore: bump version for next release (#533) 2023-07-26 09:46:28 -04:00
Louis Fortier-Dubois 589b4503df
add wgpu readme (#531) 2023-07-25 10:44:53 -04:00
Louis Fortier-Dubois 7154bde53a
patch tanh bug on mac os (#520) 2023-07-24 19:29:10 -04:00
Nathaniel Simard eaef215b17
Feat: wgpu cast tensor type (#515) 2023-07-24 11:50:44 -04:00
Louis Fortier-Dubois 9aca1837c2
Example/wgpu/mnist (#514) 
* add wgpu for mnist

* auto graphics api

* fix display tests

* clipy
 2023-07-20 17:12:13 -04:00
Nathaniel Simard d7ce52f0da
Feat/wgpu/conv (#512) 2023-07-20 15:14:42 -04:00
Nathaniel Simard 3a153d5bd0
Feat/wgpu/default device (#513) 2023-07-20 15:07:50 -04:00
Louis Fortier-Dubois df2f1492f8
Feat/wgpu/matmul transpose (#509) 2023-07-20 14:21:58 -04:00
Louis Fortier-Dubois 4b60c0e7a0
continuous to contiguous (#511) 2023-07-20 11:28:35 -04:00
Louis Fortier-Dubois 57a5476c89
bugfix for macos test (#503) 2023-07-18 16:15:00 -04:00
Louis Fortier-Dubois 5ece894e02
Bugfix/matmul/asymetric shapes (#504) 2023-07-18 16:14:43 -04:00
Nathaniel Simard f7c7d35ef5
Feat/wgpu/avg pooling (#502) 2023-07-18 11:36:57 -04:00
Nathaniel Simard c4afff182f
Feat/wgpu/max pool2d (#500) 2023-07-14 13:58:08 -04:00
Dilshod Tadjibaev e267fc1e6f
Temporarily disable broken tests for M1 Mac (#491) 
A temp workaround for #480
 2023-07-13 17:10:26 -04:00
Dilshod Tadjibaev 53c088209d
Fix new clippy warnings that cause the CI to fail (#494) 2023-07-13 13:39:39 -04:00
Nathaniel Simard a2ac2057d8
Fix: wgpu scatter with different shapes (#489) 2023-07-12 13:21:19 -04:00
Nathaniel Simard ddbbe39d74
Fix: wgpu cat + copy ops (#477) 2023-07-07 12:19:10 -04:00
Nathaniel Simard 513b9281c2
Feat/matmul/faster (#479) 2023-07-07 12:00:37 -04:00
Nathaniel Simard 261aa952c0
Add data benchmark (#474) 2023-07-07 10:21:24 -04:00
Nathaniel Simard 017485b9ea
Refactor/wgpu/ops (#472) 2023-07-06 14:26:27 -04:00
Nathaniel Simard 04ad14a32a
refactor: wgpu reductions (#471) 2023-07-06 11:40:37 -04:00
Dilshod Tadjibaev e62ee1269b
Fix burn-tch's random implementation for standard dist (#469) 2023-07-06 08:50:50 -04:00
Nathaniel Simard 65bf6c1cbb
Refactor index => slice (#466) 2023-07-05 16:30:11 -04:00
Louis Fortier-Dubois 042d2201d2
Feat/wgpu/matmul benches (#465) 2023-07-04 09:34:42 -04:00
Nathaniel Simard 98d4abd9c0
Perf/wgpu/comparison (#460) 2023-07-04 09:33:05 -04:00
Louis Fortier-Dubois 57ae13c1c0
Perf/wgpu/matmul tests (#458) 2023-07-02 16:09:24 -04:00
Nathaniel Simard 695d30918f
fix: wgpu cat can support bigger tensors (#456) 2023-07-02 12:59:32 -04:00
Nathaniel Simard a1c9970373
Refactor/wgpu/mask (#453) 2023-07-02 12:59:22 -04:00
Nathaniel Simard 5341c0260b
Perf/wgpu/benches (#448) 2023-07-02 12:57:15 -04:00
Nathaniel Simard 77219f7262
Perf/unary binary (#446) 2023-06-29 08:56:35 -04:00
Louis Fortier-Dubois f99fe0fadd
Matmul 2D Tiling (#442) 2023-06-28 16:48:15 -04:00
Nathaniel Simard 13836ac599
Fix: wgpu powf (#443) 2023-06-28 12:50:16 -04:00
Nathaniel Simard d26dead299
fix arange (#440) 2023-06-27 12:47:49 -04:00
Nathaniel Simard 4ab3ab6554
Feat/wgpu/gather scatter (#436) 2023-06-27 08:02:47 -04:00
Nathaniel Simard b6be22b855
Simpler CI (#407) 2023-06-26 08:58:54 -04:00
Nathaniel Simard 6c834d3b22
Feat/wgpu/index select (#425) 2023-06-23 09:31:49 -04:00