NVIDIA / NVIDIA/TransformerEngine

whl contains .cpp files

Open
#1,893 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

build question
Dominant language
Python
Stars
3.5k
Forks
831
Avg merge
3d 11h
Merged PRs (30d)
65

Description

whl package contains .cpp files in pytorch/csrc. Is that code need compile on target machine?

/usr/local/lib/python3.12/dist-packages/transformer_engine# ls pytorch/csrc/extensions/
activation.cpp  attention.cu  cast.cpp               gemm.cpp  multi_tensor       nvshmem_comm.cpp  permutation.cu  quantizer.cpp  softmax.cpp  transpose.cpp
apply_rope.cpp  bias.cpp      comm_gemm_overlap.cpp  misc.cpp  normalization.cpp  padding.cpp       pybind.cpp      recipe.cpp     swizzle.cpp  type_converters.cpp

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by examining the installed wheel contents under pytorch/csrc/extensions/ and tracing the project’s packaging entry points. Determine whether the .cpp files are required at runtime or were included unintentionally, then verify the expected wheel contents with a local build and inspection.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, python, pytorch
Domain
build-system
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.