NVIDIA / NVIDIA/TransformerEngine
whl contains .cpp files
Open
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
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- 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