Skip to content

enable float8 CI on sm89 #575

@vkuzo

Description

@vkuzo

Float8 was moved to torchao in #551, and currently the CI that we have for float8 is running on:
a. CPU nightly (skips all cuda related tests)
b. CUDA nightly (skips all cuda related tests which require torch._scaled_mm, because the default machines used for this do not have a high enough CUDA capability version.

We should enable float8 CI on sm89 machines, which have cuda capability 8.9. The performance will not be representative, but we can at least test correctness.

Pointers:

# old
is_H100 = torch.cuda.is_available() and torch.cuda.get_device_capability() >= (9, 0)

# new
is_cuda_8_9 = torch.cuda.is_available() and torch.cuda.get_device_capability() >= (8, 9)

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions