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

[PyTorch][Inference][ARM64] Add GPU deep canary tests + minor fixes #4455

Merged
merged 4 commits into from
Dec 5, 2024

Conversation

sallyseok
Copy link
Contributor

@sallyseok sallyseok commented Dec 5, 2024

GitHub Issue #, if available:

Note:

  • If merging this PR should also close the associated Issue, please also add that Issue # to the Linked Issues section on the right.

  • All PR's are checked weekly for staleness. This PR will be closed if not updated in 30 days.

Description

  • Added tests test_ec2_pytorch_inference_graviton_gpu_deep_canary and test_ec2_pytorch_inference_arm64_gpu_deep_canary

Tests run

  • Tests test_ec2_pytorch_inference_arm64_cpu_deep_canary and test_ec2_pytorch_inference_arm64_gpu_deep_canary passed on 3feea56 with arm64_mode = true, deep_canary_mode = true, and build_frameworks = ["pytorch"].
  • Tests with graviton_mode = true passed on 584c748

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license. I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@sallyseok sallyseok requested a review from a team as a code owner December 5, 2024 21:31
@aws-deep-learning-containers-ci aws-deep-learning-containers-ci bot added the Size:XS Determines the size of the PR label Dec 5, 2024
@sallyseok sallyseok requested a review from a team as a code owner December 5, 2024 22:14
@aws-deep-learning-containers-ci aws-deep-learning-containers-ci bot added ec2 Reflects file change in dlc_tests/ec2 folder Size:S Determines the size of the PR test Reflects file change in test folder labels Dec 5, 2024
@sallyseok sallyseok requested review from a team as code owners December 5, 2024 23:05
@sallyseok sallyseok changed the title Test deep canary for PT arm64 [PyTorch][Inference][ARM64] Add GPU deep canary tests + minor fixes Dec 5, 2024
@sallyseok sallyseok enabled auto-merge (squash) December 5, 2024 23:07
@sallyseok sallyseok merged commit eab6156 into aws:master Dec 5, 2024
30 checks passed
@sallyseok sallyseok deleted the test-deep-canary branch December 5, 2024 23:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ec2 Reflects file change in dlc_tests/ec2 folder Size:S Determines the size of the PR Size:XS Determines the size of the PR test Reflects file change in test folder
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants