Add one AVX target

This commit is contained in:
Henry Case 2021-02-17 13:38:23 +00:00
parent 67f275c898
commit 424bd89890

View File

@ -13,21 +13,21 @@ set(CMAKE_CXX_STANDARD 11)
string(TOLOWER "${CMAKE_BUILD_TYPE}" CMAKE_BUILD_TYPE_LOWER)
if(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "x86_64")
set(ARCH "QRS_ARCH_x86_64")
set(ARCH "ARCH_x86_64")
elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "amd64")
set(ARCH "QRS_ARCH_x86_64")
set(ARCH "ARCH_x86_64")
elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "AMD64")
set(ARCH "QRS_ARCH_x86_64")
set(ARCH "ARCH_x86_64")
elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "x86")
set(ARCH "QRS_ARCH_x86")
set(ARCH "ARCH_x86")
elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "i386")
set(ARCH "QRS_ARCH_x86")
set(ARCH "ARCH_x86")
elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "i686")
set(ARCH "QRS_ARCH_x86")
set(ARCH "ARCH_x86")
elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "aarch64")
set(ARCH "QRS_ARCH_aarch64")
set(ARCH "ARCH_aarch64")
elseif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "arm64")
set(ARCH "QRS_ARCH_aarch64")
set(ARCH "ARCH_aarch64")
else()
message(FATAL_ERROR "Unknown processor:" ${CMAKE_SYSTEM_PROCESSOR})
endif()
@ -82,16 +82,8 @@ include_directories(
set(CMAKE_C_FLAGS "${C_CXX_FLAGS} -D${ARCH}")
set(CMAKE_CXX_FLAGS "${C_CXX_FLAGS} -D${ARCH}")
add_library(
common
OBJECT
common/fips202.c
common/sp800-185.c
common/randombytes.c
)
function(define_crypto_alg name namespace src inc)
# Common function for defining algorithm component
function(define_crypto_alg name namespace src inc test_src)
add_library(
pqclean_${name}
OBJECT
@ -101,6 +93,7 @@ function(define_crypto_alg name namespace src inc)
target_include_directories(
pqclean_${name} PRIVATE
common
${inc}
)
target_compile_definitions(
@ -111,7 +104,7 @@ function(define_crypto_alg name namespace src inc)
add_library(
pqclean_test_${name}
OBJECT
test/crypto_sign/testvectors.c
${test_src}
)
target_compile_definitions(
@ -124,8 +117,27 @@ function(define_crypto_alg name namespace src inc)
common
${inc}
)
add_executable(
test_runner_${name}
)
target_link_libraries(
test_runner_${name}
common
pqclean_${name}
pqclean_test_${name}
)
endfunction()
function(define_kem_alg name namespace src inc)
define_crypto_alg(${name} ${namespace} "${src}" "${inc}" test/crypto_kem/testvectors.c)
endfunction()
function(define_sig_alg name namespace src inc)
define_crypto_alg(${name} ${namespace} "${src}" "${inc}" test/crypto_sign/testvectors.c)
endfunction()
# Define sources of the components
set(
SRC_CLEAN_DILITHIUM2
crypto_sign/dilithium2/clean/ntt.c
@ -142,10 +154,117 @@ set(
INC_CLEAN_DILITHIUM2
crypto_sign/dilithium2/clean
)
define_crypto_alg(dilithium2_clean
define_sig_alg(dilithium2_clean
PQCLEAN_DILITHIUM2_CLEAN "${SRC_CLEAN_DILITHIUM2}" "${INC_CLEAN_DILITHIUM2}")
set(
SRC_CLEAN_DILITHIUM3
crypto_sign/dilithium3/clean/ntt.c
crypto_sign/dilithium3/clean/packing.c
crypto_sign/dilithium3/clean/poly.c
crypto_sign/dilithium3/clean/polyvec.c
crypto_sign/dilithium3/clean/reduce.c
crypto_sign/dilithium3/clean/rounding.c
crypto_sign/dilithium3/clean/sign.c
crypto_sign/dilithium3/clean/symmetric-shake.c
)
set(
INC_CLEAN_DILITHIUM3
crypto_sign/dilithium3/clean
)
define_sig_alg(dilithium3_clean
PQCLEAN_DILITHIUM3_CLEAN "${SRC_CLEAN_DILITHIUM3}" "${INC_CLEAN_DILITHIUM3}")
set(
SRC_CLEAN_DILITHIUM5
crypto_sign/dilithium5/clean/ntt.c
crypto_sign/dilithium5/clean/packing.c
crypto_sign/dilithium5/clean/poly.c
crypto_sign/dilithium5/clean/polyvec.c
crypto_sign/dilithium5/clean/reduce.c
crypto_sign/dilithium5/clean/rounding.c
crypto_sign/dilithium5/clean/sign.c
crypto_sign/dilithium5/clean/symmetric-shake.c
)
set(
INC_CLEAN_DILITHIUM5
crypto_sign/dilithium5/clean
)
define_sig_alg(dilithium5_clean
PQCLEAN_DILITHIUM5_CLEAN "${SRC_CLEAN_DILITHIUM5}" "${INC_CLEAN_DILITHIUM5}")
set(
SRC_CLEAN_KYBER512
crypto_kem/kyber512/clean/cbd.c
crypto_kem/kyber512/clean/indcpa.c
crypto_kem/kyber512/clean/kem.c
crypto_kem/kyber512/clean/ntt.c
crypto_kem/kyber512/clean/poly.c
crypto_kem/kyber512/clean/polyvec.c
crypto_kem/kyber512/clean/reduce.c
crypto_kem/kyber512/clean/symmetric-shake.c
crypto_kem/kyber512/clean/verify.c
)
set(
INC_CLEAN_KYBER512
crypto_kem/kyber512/clean
)
define_kem_alg(kyber512_clean
PQCLEAN_KYBER512_CLEAN "${SRC_CLEAN_KYBER512}" "${INC_CLEAN_KYBER512}")
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -march=haswell")
set(
SRC_AVX2_KYBER512
crypto_kem/kyber512/avx2/cbd.c
crypto_kem/kyber512/avx2/consts.c
crypto_kem/kyber512/avx2/fips202x4.c
crypto_kem/kyber512/avx2/indcpa.c
crypto_kem/kyber512/avx2/kem.c
crypto_kem/kyber512/avx2/poly.c
crypto_kem/kyber512/avx2/polyvec.c
crypto_kem/kyber512/avx2/rejsample.c
crypto_kem/kyber512/avx2/symmetric-shake.c
crypto_kem/kyber512/avx2/verify.c
crypto_kem/kyber512/avx2/basemul.S
crypto_kem/kyber512/avx2/fq.S
crypto_kem/kyber512/avx2/invntt.S
crypto_kem/kyber512/avx2/ntt.S
crypto_kem/kyber512/avx2/shuffle.S
)
if(${ARCH} STREQUAL "ARCH_x86_64")
set(
INC_AVX2_KYBER512
crypto_kem/kyber512/avx2
)
define_kem_alg(kyber512_avx2
PQCLEAN_KYBER512_AVX2 "${SRC_AVX2_KYBER512}" "${INC_AVX2_KYBER512}")
endif()
# The rest of the library
set(SRC_COMMON_GENERIC
common/fips202.c
common/sp800-185.c
common/randombytes.c
)
if(${ARCH} STREQUAL "ARCH_x86_64")
set(SRC_COMMON_AVX2
common/keccak4x/KeccakP-1600-times4-SIMD256.c
)
endif()
add_library(
common
OBJECT
${SRC_COMMON_GENERIC}
${SRC_COMMON_AVX2}
)
add_library(
pqclean
SHARED
@ -160,20 +279,27 @@ target_link_libraries(
pqclean
common
pqclean_dilithium2_clean
pqclean_dilithium3_clean
pqclean_dilithium5_clean
)
target_link_libraries(
pqclean_s
common
pqclean_dilithium2_clean
pqclean_dilithium3_clean
pqclean_dilithium5_clean
)
add_executable(
test
)
target_link_libraries(
test
pqclean_s
pqclean_test_dilithium2_clean
)
# TODO: this requires changes to testvectors.c
# add_executable(
# test
# )
#
# target_link_libraries(
# test
# pqclean_s
# pqclean_test_dilithium2_clean
# pqclean_test_dilithium3_clean
# pqclean_test_dilithium5_clean
# )