Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

IPU Integration 5/5 #7867

Merged
merged 69 commits into from
Jun 11, 2021
Merged
Changes from 1 commit
Commits
Show all changes
69 commits
Select commit Hold shift + click to select a range
f75f445
Initial changes
Feb 22, 2021
a4a60c2
Merge branch 'master' into wip/acc
Mar 24, 2021
dc9744b
Add broken example for now
Mar 24, 2021
931bb74
Fix reference
Apr 7, 2021
9b18baf
Merge branch 'master' into wip/acc
May 11, 2021
c617f02
Fix format
May 11, 2021
522a81f
Code runs
May 11, 2021
0c00360
Fixes
May 26, 2021
30c1370
Merge branch 'master' into wip/acc
May 26, 2021
adbdb2a
Clear up files
May 26, 2021
3e733af
Add tests, helpers, fixes
May 27, 2021
a51f23e
Small cleanups
May 27, 2021
be7de87
Refactors based on review
Jun 1, 2021
83c8a79
Swap to special tests
Jun 1, 2021
a6018e5
Add special tests
Jun 1, 2021
0e71bbe
Add source
Jun 1, 2021
6e38bd1
Cleanups
Jun 1, 2021
526383f
Add logic to attach/detach model from devices
Jun 2, 2021
e18039c
Fixes for tests
Jun 2, 2021
2e43fee
Fixes for tests
Jun 2, 2021
53d31a0
Move earlier
Jun 2, 2021
6241432
Cleanups
Jun 2, 2021
d249a13
Add check for nvcc
Jun 2, 2021
d08cf39
Add tests, cleanups
Jun 2, 2021
7469744
Fix errors
Jun 3, 2021
f474c5b
fix
Jun 3, 2021
e178d5f
Try condition
Jun 3, 2021
c704920
Add missing annotation
Jun 3, 2021
c54a216
Clearer
Jun 3, 2021
2ea1766
Clearer message
Jun 3, 2021
751f0ea
Fix variable
Jun 3, 2021
87e4c8a
Merge branch 'master' into wip/acc
Jun 7, 2021
61d2014
Cleanups
Jun 7, 2021
d76f491
Merge branch 'master' into wip/acc
Jun 7, 2021
62860ff
Add comment
Jun 7, 2021
b5a5032
CHANGELOG.md
Jun 7, 2021
72ed367
Add simple selection test
Jun 7, 2021
88fba4a
Merge branch 'master' into wip/acc
Jun 7, 2021
3fb031d
Remove special=True to see what happens
Jun 7, 2021
515d491
Fix test
Jun 7, 2021
ed16808
Update tests/accelerators/test_ipu.py
Jun 7, 2021
7f50295
Convert ipu_cores -> ipus
Jun 7, 2021
c53cf88
Add typing, fail earlier
Jun 7, 2021
a6dbd8a
simplify precision
Jun 7, 2021
953454b
Add test, add helper
Jun 8, 2021
24829bf
fix accum
Jun 8, 2021
d7d38c5
Update pytorch_lightning/plugins/training_type/ipu.py
Jun 8, 2021
c333e27
Use stages
Jun 8, 2021
9d3741a
Make sure warning message returned
Jun 8, 2021
fd1899a
thorw error
Jun 8, 2021
0727954
Add more tests, use fs
Jun 8, 2021
ce182f7
add comment
Jun 8, 2021
7e81bcd
Clean
Jun 8, 2021
d1788d1
Address feedback, add IPU tests
Jun 9, 2021
08e5338
Fixes
Jun 9, 2021
45dc6a6
Fix signature
Jun 9, 2021
de040c6
Add types
Jun 9, 2021
42d7ab0
Remove autoround
Jun 9, 2021
8ab62c4
Merge branch 'master' into wip/acc
Jun 9, 2021
36f3672
Add docstring
Jun 9, 2021
5f89714
Merge branch 'master' into wip/acc
Jun 10, 2021
d0f98f3
Merge branch 'master' into wip/acc
Jun 10, 2021
f9d61c5
ipu_cores -> ipus
Jun 10, 2021
cf48ff8
Add test, remove unnecessary precision set
Jun 11, 2021
02a75b5
Add optimizer test
Jun 11, 2021
d18fc55
Add precision back with test
Jun 11, 2021
043884a
Address code review
Jun 11, 2021
b249391
Change to probs
Jun 11, 2021
b0dd206
Move some of the asserts earlier
Jun 11, 2021
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
CHANGELOG.md
SeanNaren committed Jun 7, 2021
commit b5a50325a739f5b6e7ff33d0e49b02babe47c59c
3 changes: 3 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -65,6 +65,9 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
- Added trainer stage hooks for Training Plugins and Accelerators ([#7864](https://github.com/PyTorchLightning/pytorch-lightning/pull/7864))


- Added IPU Accelerator ([#7867](https://github.com/PyTorchLightning/pytorch-lightning/pull/7867))


### Changed

- Changed calling of `untoggle_optimizer(opt_idx)` out of the closure function ([#7563](https://github.com/PyTorchLightning/pytorch-lightning/pull/7563)