-
Notifications
You must be signed in to change notification settings - Fork 35
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
feat: add Python implementation of accelerated OP #67
Conversation
381c658
to
42bbbd6
Compare
Codecov ReportBase: 64.46% // Head: 63.57% // Decreases project coverage by
Additional details and impacted files@@ Coverage Diff @@
## main #67 +/- ##
==========================================
- Coverage 64.46% 63.57% -0.89%
==========================================
Files 66 68 +2
Lines 2842 2883 +41
==========================================
+ Hits 1832 1833 +1
- Misses 1010 1050 +40
Flags with carried forward coverage won't be shown. Click here to find out more.
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. ☔ View full report at Codecov. |
57ad305
to
5e35102
Compare
5e35102
to
abdb3db
Compare
47ddf71
to
e4cc211
Compare
eca7bdd
to
ac2f0db
Compare
e4cc211
to
3da6046
Compare
3da6046
to
ab55926
Compare
a005547
to
42a6a80
Compare
13828d6
to
4e0f36e
Compare
84a2f9e
to
09946f1
Compare
cd4cd7b
to
1183487
Compare
1183487
to
f696543
Compare
0aee719
to
e0c7e9c
Compare
1da6716
to
4d30221
Compare
4d30221
to
ecd4a43
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
dmu: torch.Tensor, updates: torch.Tensor, mu: torch.Tensor, b1: float | ||
) -> Tuple[torch.Tensor, torch.Tensor]: | ||
"""Adam backward mu.""" | ||
dupdates = dmu.mul(1.0 - b1) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
1.0 -> double or float or?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It will be cast to the same floating point type as dmu.dtype
, usually torch.float32
.
3cea69d
to
044ffd7
Compare
044ffd7
to
8653f7b
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
8888634
to
98915c2
Compare
98915c2
to
75e8957
Compare
Description
Resolves #36
Resolves #66
Motivation and Context
Types of changes
What types of changes does your code introduce? Put an
x
in all the boxes that apply:Checklist
Go over all the following points, and put an
x
in all the boxes that apply.If you are unsure about any of these, don't hesitate to ask. We are here to help!
make format
(required)make lint
(required)make test
pass. (required)