Skip to content

[Question] Confused about the assignment logic for device_capability. #35

@yuecheng-ma

Description

@yuecheng-ma

device_capability = ""

When TORCH_CUDA_ARCH_LIST is not empty, set device_capability to empty. Shouldn't TORCH_CUDA_ARCH_LIST be assigned to device_capability ? Otherwise, the specification of TORCH_CUDA_ARCH_LIST would be invalid.

$ TORCH_CUDA_ARCH_LIST=8.0 GROUPED_GEMM_CUTLASS=1 pip install .

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions