Skip to content

FBGEMM cutlass used instead of PyTorch third_party/cutlass #165110

@danielvegamyhre

Description

@danielvegamyhre

🐛 Describe the bug

Repro from @eqy

  1. git clone https://github.com/pytorch/pytorch
  2. TORCH_CUDA_ARCH_LIST=10.0 python setup.py develop
  3. open third_party/cutlass/include/cutlass/arch/arch.h for editing,
    make random invalid changes to the file so it won't compile
  4. TORCH_CUDA_ARCH_LIST=10.0 python setup.py develop
    observe the build completes normally, when it should fail

Alternatively, try to switch one of the ArchTags in aten/src/ATen/native/cuda/RowwiseScaledMM.cu to Sm103 and observe the build breaks complaining that Sm103 isn't a member of cutlass::arch when it's clearly present in third_party/cutlass/include/cutlass/arch/arch.h

Versions

latest pytorch main branch, built from source

cc @malfet @seemethere

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: buildBuild system issuesmodule: third_partytriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions