Skip to content

Updates to LBANN to support ROCm-enabled DistConv#2145

Merged
benson31 merged 6 commits into
LBANN:developfrom
benson31:update-rocm-distconv
Nov 12, 2022
Merged

Updates to LBANN to support ROCm-enabled DistConv#2145
benson31 merged 6 commits into
LBANN:developfrom
benson31:update-rocm-distconv

Conversation

@benson31

Copy link
Copy Markdown
Collaborator

Depends on LBANN/DiHydrogen#65.

@KIwabuchi KIwabuchi left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Looks good to me!

@bvanessen bvanessen left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

LGTM. CI unit tests pass on Tioga.

Comment thread src/callbacks/check_gradients.cpp Outdated
@szaman19 szaman19 mentioned this pull request Nov 9, 2022
1 task
@benson31
benson31 force-pushed the update-rocm-distconv branch from c41c701 to 547814e Compare November 11, 2022 17:24
@benson31
benson31 merged commit 2e3012a into LBANN:develop Nov 12, 2022
@benson31
benson31 deleted the update-rocm-distconv branch November 12, 2022 00:17
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.

3 participants