Skip to content

Commit 4585206

Browse files
tanquermeta-codesync[bot]
authored andcommitted
Fix oss aarch64 build with transport (#71)
Summary: Pull Request resolved: #71 Ctran requires higher cuda arch. Get rid of transport option for docs building. Reviewed By: d4l3k Differential Revision: D88040729 fbshipit-source-id: e86859fa27c666e69724909064622fe5a01637bf
1 parent 4d56e11 commit 4585206

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

.github/workflows/docs.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ jobs:
4949
pip install -r docs/requirements.txt
5050
export USE_NCCL=0
5151
export USE_NCCLX=0
52+
export USE_TRANSPORT=0
5253
export USE_SYSTEM_LIBS=1
5354
pip install --no-build-isolation .[dev] -v
5455
- name: Build Sphinx Docs

comms/ctran/CMakeLists.txt

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,13 @@ if (NOT DEFINED ENABLE_TCPDM)
1313
list(FILTER CTRAN_SOURCES EXCLUDE REGEX ".*/ctran/backends/tcpdevmem/.*")
1414
endif()
1515

16+
list(FILTER CMAKE_CUDA_FLAGS EXCLUDE REGEX ".*arch=compute_50,code=sm_50")
17+
list(FILTER CMAKE_CUDA_FLAGS EXCLUDE REGEX ".*arch=compute_60,code=sm_60")
18+
list(FILTER CMAKE_CUDA_FLAGS EXCLUDE REGEX ".*arch=compute_70,code=sm_70")
19+
list(FILTER CMAKE_CUDA_FLAGS EXCLUDE REGEX ".*arch=compute_75,code=sm_75")
20+
21+
message(status "CMAKE_CUDA_FLAGS: ${CMAKE_CUDA_FLAGS}")
22+
1623
# genctran.py generates source files (*.cu) for template instantiations across
1724
# different algorithms, data types, and operations.
1825
execute_process(

0 commit comments

Comments
 (0)