8000 Enable GB200 for dynamo/torchbench tests · Issue #154560 · pytorch/pytorch · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
Enable GB200 for dynamo/torchbench tests #154560
Open
@zhe-thoughts

Description

@zhe-thoughts

🚀 The feature, motivation and pitch

We should enable PyTorch CI jobs on GB200 overall.

This is to track a subset of the work: running dynamo torchbench on GB200 https://github.com/pytorch/pytorch/blob/main/benchmarks/dynamo/README.md

Alternatives

No response

Additional context

No response

cc @ptrblck @msaroufim @eqy @jerryzh168 @seemethere @malfet @pytorch/pytorch-dev-infra @chauhang @penguinwu

Metadata

Metadata

Assignees

Labels

featureA request for a proper, new feature.module: ciRelated to continuous integrationmodule: cudaRelated to torch.cuda, and CUDA support in generaloncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

Status

In Progress

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions

    0