File tree Expand file tree Collapse file tree 7 files changed +1
-1360
lines changed Expand file tree Collapse file tree 7 files changed +1
-1360
lines changed Original file line number Diff line number Diff line change 1- <<<<<<< HEAD
211cc4a7c4aac921b7fa4658e162058632b7faed55
3- =======
4- 27664085f804afc83df26f740bb46c365854f2c4
5- >>>>>>> upstream/main
Original file line number Diff line number Diff line change @@ -46,15 +46,11 @@ case ${DOCKER_TAG_PREFIX} in
4646 BASE_TARGET=rocm
4747 GPU_IMAGE=rocm/dev-ubuntu-22.04:${GPU_ARCH_VERSION} -complete
4848 PYTORCH_ROCM_ARCH=" gfx900;gfx906;gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1102;gfx1200;gfx1201"
49- <<< <<< < HEAD
50- DOCKER_GPU_BUILD_ARG=" --build-arg PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH} "
51- =======
5249 # add gfx950 conditionally starting in ROCm 7.0
5350 if [[ " $GPU_ARCH_VERSION " == * " 7.0" * ]]; then
5451 PYTORCH_ROCM_ARCH=" ${PYTORCH_ROCM_ARCH} ;gfx950"
5552 fi
5653 DOCKER_GPU_BUILD_ARG=" --build-arg PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH} --build-arg ROCM_VERSION=${GPU_ARCH_VERSION} "
57- >>>>>>> upstream/main
5854 ;;
5955 * )
6056 echo " ERROR: Unrecognized DOCKER_TAG_PREFIX: ${DOCKER_TAG_PREFIX} "
Original file line number Diff line number Diff line change @@ -903,13 +903,8 @@ cmake_dependent_option(
903903 USE_FBGEMM_GENAI
904904 "Whether to build FBGEMM GenAI quantized GEMM kernels.\
905905 Will be disabled if not supported by the platform"
906- <<<<<<< HEAD
907- OFF
908- "USE_CUDA OR USE_ROCM"
909- =======
910906 ${USE_FBGEMM_GENAI_DEFAULT}
911907 "(USE_CUDA AND NOT MSVC) OR USE_ROCM"
912- >>>>>>> upstream/main
913908 OFF )
914909
915910
Original file line number Diff line number Diff line change 11# Build System requirements
2- <<<<<<< HEAD
32setuptools>=70.1.0,<80.0 # setuptools develop deprecated on 80.0
43cmake>=3.31.4
54ninja==1.11.1.3
@@ -10,15 +9,4 @@ pyyaml==6.0.2
109requests==2.32.4
1110six==1.17.0 # dependency chain: NNPACK -> PeachPy -> six
1211typing-extensions==4.14.1
13- =======
14- setuptools>=70.1.0
15- cmake>=3.27
16- ninja
17- numpy
18- packaging
19- pyyaml
20- requests
21- six # dependency chain: NNPACK -> PeachPy -> six
22- typing-extensions>=4.10.0
23- >>>>>>> upstream/main
2412pip # not technically needed, but this makes setup.py invocation work
You can’t perform that action at this time.
0 commit comments