dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
735 Commits 39 Branches 19 Tags 101 MiB
Commit Graph

113 Commits

Author SHA1 Message Date
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
Dilshod Tadjibaev f0b266c8a3
Add missing docs for burn-wgpu and burn-tch (#427) 2023-06-23 09:31:37 -04:00
Nathaniel Simard c4e4c25fef
Feat/wgpu/mask (#422) 2023-06-21 13:01:18 -04:00
Dilshod Tadjibaev fce45f51be
Doc fixes (#418) 2023-06-21 12:32:50 -04:00
Nathaniel Simard b6b684b9f2
Feat/wgpu/cat (#416) 2023-06-20 16:16:10 -04:00
Nathaniel Simard 4d40bde7b9
feat: argmax + argmin (#412) 2023-06-20 10:03:00 -04:00
Nathaniel Simard 323261b594
Feat/wgpu/comparisons (#411) 2023-06-19 19:11:49 -04:00
Nathaniel Simard a8624590af
feat: mask_where (#409) 2023-06-18 15:04:28 -04:00
Nathaniel Simard b6f88f8fca
refactor: better templating for wgsl shaders (#404) 2023-06-15 11:59:37 -04:00
Nathaniel Simard 18b7d5202d
Fix wgpu context creation (#405) 2023-06-15 11:57:56 -04:00
Dilshod Tadjibaev 834c7ecc1f
Clean up cargo descriptions and formatting (#403) 2023-06-15 09:20:53 -04:00
Nathaniel Simard 8ed0775dfd
Perf/wgpu/context (#395) 2023-06-12 14:24:19 -04:00
Nathaniel Simard 8c9802c363
Feat/wgpu/reduction (#392) 2023-06-08 16:54:36 -04:00
Nathaniel Simard bff752b1a8
Feat/wgpu/index (#386) 2023-06-06 12:29:14 -04:00
Nathaniel Simard c1e1e38a79
Add more ops (#387) 2023-06-06 12:21:20 -04:00
Nathaniel Simard ecc67c58f9
Feat/wgpu/swap dims (#381) 2023-06-04 19:34:35 -04:00
Nathaniel Simard 0a205a3603
feat: add basic matmul for wgpu backend (#379) 2023-06-03 10:32:57 -04:00
Nathaniel Simard 974fdfaba1
Feat/wgpu backend setup (#376) 2023-06-02 11:52:47 -04:00