Skip to content

GH200 support#369

Merged
gagank1 merged 24 commits into
mainfrom
gkaushik/bionemo2-gh200
Dec 10, 2024
Merged

GH200 support#369
gagank1 merged 24 commits into
mainfrom
gkaushik/bionemo2-gh200

Conversation

@gagank1
Copy link
Copy Markdown
Contributor

@gagank1 gagank1 commented Oct 28, 2024

Adds support for GH200 machines.

Summary

Most of the differences between Dockerfile.arm and Dockerfile involve building packages that are either not available on ARM or don't work for some reason the way they are available on ARM.

Usage

Usage is the same as the normal framework container. It will build and run on GH200 machines now.

Testing

All the existing tests pass on GH200 machines (except for this known bug which results in 1 failure).

Tests for these changes can be run via:
./ci/scripts/run_pytest.sh

@gagank1 gagank1 marked this pull request as ready for review December 9, 2024 07:39
@gagank1 gagank1 requested review from trvachov and yzhang123 December 9, 2024 07:41
@gagank1 gagank1 changed the title Draft: GH200 support GH200 support Dec 9, 2024
@gagank1 gagank1 self-assigned this Dec 9, 2024
Copy link
Copy Markdown
Collaborator

@trvachov trvachov 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 -- I asked a few questions to just understand where things are coming from. Thanks!

Comment thread Dockerfile.arm
Comment thread Dockerfile.arm
Comment thread Dockerfile.arm
Comment thread Dockerfile.arm
Comment thread Dockerfile.arm
Comment thread Dockerfile.arm Outdated
Comment thread Dockerfile.arm Outdated
@gagank1 gagank1 removed the request for review from yzhang123 December 10, 2024 20:00
@gagank1
Copy link
Copy Markdown
Contributor Author

gagank1 commented Dec 10, 2024

/build-ci

@gagank1 gagank1 merged commit 462f65e into main Dec 10, 2024
@gagank1 gagank1 deleted the gkaushik/bionemo2-gh200 branch December 10, 2024 23:57
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants