Skip to content

Commit b5b182e

Browse files
committed
Revert "Adding cpu inference with VXE ISA for s390x architecture (#12613)"
This reverts commit ada1921.
1 parent 0578e5a commit b5b182e

File tree

7 files changed

+8
-653
lines changed

7 files changed

+8
-653
lines changed

Dockerfile.s390x

Lines changed: 0 additions & 152 deletions
This file was deleted.

cmake/cpu_extension.cmake

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,6 @@ else()
8181
find_isa(${CPUINFO} "POWER9" POWER9_FOUND)
8282
find_isa(${CPUINFO} "asimd" ASIMD_FOUND) # Check for ARM NEON support
8383
find_isa(${CPUINFO} "bf16" ARM_BF16_FOUND) # Check for ARM BF16 support
84-
find_isa(${CPUINFO} "S390" S390_FOUND)
8584
endif()
8685

8786

@@ -130,16 +129,8 @@ elseif (ASIMD_FOUND)
130129
elseif(APPLE_SILICON_FOUND)
131130
message(STATUS "Apple Silicon Detected")
132131
set(ENABLE_NUMA OFF)
133-
elseif (S390_FOUND)
134-
message(STATUS "S390 detected")
135-
# Check for S390 VXE support
136-
list(APPEND CXX_COMPILE_FLAGS
137-
"-mvx"
138-
"-mzvector"
139-
"-march=native"
140-
"-mtune=native")
141132
else()
142-
message(FATAL_ERROR "vLLM CPU backend requires AVX512, AVX2, Power9+ ISA, S390X ISA or ARMv8 support.")
133+
message(FATAL_ERROR "vLLM CPU backend requires AVX512, AVX2, Power9+ ISA or ARMv8 support.")
143134
endif()
144135

145136
#

csrc/cpu/attention.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,8 @@ struct KernelVecType<float> {
2424

2525
template <>
2626
struct KernelVecType<c10::Half> {
27-
#if defined(__powerpc64__) || defined(__s390x__)
28-
// Power and s390x architecture-specific vector types
27+
#ifdef __powerpc64__
28+
// Power architecture-specific vector types
2929
using q_load_vec_type = vec_op::FP32Vec8;
3030
using k_load_vec_type = vec_op::FP32Vec16;
3131
using v_load_vec_type = vec_op::FP32Vec16;

csrc/cpu/cpu_types.hpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,6 @@
77
#elif defined(__POWER9_VECTOR__)
88
// ppc implementation
99
#include "cpu_types_vsx.hpp"
10-
#elif defined(__s390x__)
11-
// s390 implementation
12-
#include "cpu_types_vxe.hpp"
1310
#elif defined(__aarch64__)
1411
// arm implementation
1512
#include "cpu_types_arm.hpp"

0 commit comments

Comments
 (0)