Skip to content

Commit

Permalink
change default platform to A55 everywhere; add missing examples
Browse files Browse the repository at this point in the history
  • Loading branch information
mkannwischer committed Mar 17, 2024
1 parent ba9ad59 commit 7476f4c
Showing 1 changed file with 14 additions and 6 deletions.
20 changes: 14 additions & 6 deletions example.py
Original file line number Diff line number Diff line change
Expand Up @@ -550,7 +550,7 @@ def core(self, slothy):


class ntt_kyber_1234_567(Example):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA72, timeout=None):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA55, timeout=None):
name = "ntt_kyber_1234_567"
infile = name

Expand Down Expand Up @@ -598,7 +598,7 @@ def core(self, slothy):


class ntt_kyber_1234(Example):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA72):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA55):
name = "ntt_kyber_1234"
infile = "ntt_kyber_1234_567"

Expand All @@ -623,7 +623,7 @@ def core(self, slothy):


class ntt_kyber_567(Example):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA72, timeout=None):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA55, timeout=None):
name = "ntt_kyber_567"
infile = "ntt_kyber_1234_567"

Expand Down Expand Up @@ -976,7 +976,7 @@ def core(self, slothy):


class ntt_dilithium_1234_5678(Example):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA72, timeout=None):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA55, timeout=None):
name = f"ntt_dilithium_1234_5678"
infile = name

Expand Down Expand Up @@ -1019,7 +1019,7 @@ def core(self, slothy):


class ntt_dilithium_1234(Example):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA72):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA55):
name = "ntt_dilithium_1234"
infile = "ntt_dilithium_1234_5678"

Expand All @@ -1043,7 +1043,7 @@ def core(self, slothy):


class ntt_dilithium_5678(Example):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA72):
def __init__(self, var="", arch=AArch64_Neon, target=Target_CortexA55):
name = "ntt_dilithium_5678"
infile = "ntt_dilithium_1234_5678"

Expand Down Expand Up @@ -1174,13 +1174,17 @@ def main():
ntt_kyber_123_4567(var="scalar_load_store"),
ntt_kyber_123_4567(var="manual_st4"),
ntt_kyber_1234_567(),
ntt_kyber_1234(),
ntt_kyber_567(),
# Cortex-A72
ntt_kyber_123_4567(target=Target_CortexA72),
ntt_kyber_123_4567(var="scalar_load", target=Target_CortexA72),
ntt_kyber_123_4567(var="scalar_store", target=Target_CortexA72),
ntt_kyber_123_4567(var="scalar_load_store", target=Target_CortexA72),
ntt_kyber_123_4567(var="manual_st4", target=Target_CortexA72),
ntt_kyber_1234_567(target=Target_CortexA72),
ntt_kyber_1234(target=Target_CortexA72),
ntt_kyber_567(target=Target_CortexA72),
# # Apple M1 Firestorm
ntt_kyber_123_4567(target=Target_AppleM1_firestorm, timeout=3600),
ntt_kyber_123_4567(var="scalar_load", target=Target_AppleM1_firestorm, timeout=3600),
Expand Down Expand Up @@ -1218,12 +1222,16 @@ def main():
ntt_dilithium_123_45678(var="manual_st4"),
ntt_dilithium_1234_5678(),
ntt_dilithium_1234_5678(var="manual_st4"),
ntt_dilithium_1234(),
ntt_dilithium_5678(),
# Cortex-A72
ntt_dilithium_123_45678(target=Target_CortexA72),
ntt_dilithium_123_45678(var="w_scalar", target=Target_CortexA72),
ntt_dilithium_123_45678(var="manual_st4", target=Target_CortexA72),
ntt_dilithium_1234_5678(target=Target_CortexA72),
ntt_dilithium_1234_5678(var="manual_st4", target=Target_CortexA72),
ntt_dilithium_1234(target=Target_CortexA72),
ntt_dilithium_5678(target=Target_CortexA72),
# Apple M1 Firestorm
ntt_dilithium_123_45678(target=Target_AppleM1_firestorm, timeout=3600),
ntt_dilithium_123_45678(var="w_scalar", target=Target_AppleM1_firestorm, timeout=3600),
Expand Down

0 comments on commit 7476f4c

Please sign in to comment.