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

Add WIP docstrings to CPU and GPU #1632

Merged
merged 8 commits into from
Aug 22, 2021
Merged

Add WIP docstrings to CPU and GPU #1632

merged 8 commits into from
Aug 22, 2021

Conversation

logankilpatrick
Copy link
Member

We could surely use more details, but ideally we would want to document these functions and what they do + usage.

Copy link
Member

@DhairyaLGandhi DhairyaLGandhi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Lgtm

src/functor.jl Outdated Show resolved Hide resolved
src/functor.jl Outdated Show resolved Hide resolved
src/functor.jl Outdated Show resolved Hide resolved
DhairyaLGandhi
DhairyaLGandhi previously approved these changes Jun 26, 2021
@DhairyaLGandhi
Copy link
Member

bors r+

bors bot added a commit that referenced this pull request Jun 26, 2021
1632: Add WIP docstrings to CPU and GPU r=DhairyaLGandhi a=logankilpatrick

We could surely use more details, but ideally we would want to document these functions and what they do + usage.


Co-authored-by: Logan Kilpatrick <23kilpatrick23@gmail.com>
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
@bors
Copy link
Contributor

bors bot commented Jun 26, 2021

Canceled.

@DhairyaLGandhi
Copy link
Member

DhairyaLGandhi commented Jun 26, 2021

bors r-

Just wanted to get an opinion on whether to include details about needing a CUDA compatible GPU atm. Also, good to add an example for each, ideally with a simple model.

@ToucheSir
Copy link
Member

Given #1566, I don't think it's necessary to include details about GPU compat for CUDA specifically.

@logankilpatrick
Copy link
Member Author

Would a simple example like:

julia> m = Dense(10, 5)
Dense(10, 5)

julia> cpu(m)
Dense(10, 5)

work?

@DhairyaLGandhi
Copy link
Member

That'd work, i would also show that the types of arrays have changed to a CuArray (for gpu)

@logankilpatrick
Copy link
Member Author

I don't have an Nvidia GPU, can you add that example?

@DhairyaLGandhi
Copy link
Member

Done.

Copy link
Member

@darsnack darsnack left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just some suggestions to make it clearer and a forward link to @functor so people aren't confused when they custom layer won't transfer.

src/functor.jl Outdated Show resolved Hide resolved
src/functor.jl Outdated Show resolved Hide resolved
logankilpatrick and others added 3 commits August 22, 2021 14:10
Co-authored-by: Kyle Daruwalla <daruwalla.k.public@icloud.com>
Co-authored-by: Kyle Daruwalla <daruwalla.k.public@icloud.com>
@logankilpatrick
Copy link
Member Author

I tried to address the comments, please let me know if more needs to be updated!

@darsnack
Copy link
Member

bors r+

@bors
Copy link
Contributor

bors bot commented Aug 22, 2021

Build succeeded:

@bors bors bot merged commit 48f6ae9 into master Aug 22, 2021
@DhairyaLGandhi DhairyaLGandhi deleted the logankilpatrick-patch-1 branch August 23, 2021 06:05
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants