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.

Copy link
Copy Markdown

@KIwabuchi KIwabuchi left a comment

Choose a reason for hiding this comment

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

Looks good to me!

Copy link
Copy Markdown
Collaborator

@bvanessen bvanessen left a comment

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