cuDilithium: High-Throughput GPU Implementation of Dilithium Post-Quantum Digital Signature

Description

CUDA implementation of Dilithium, a post-quantum signature scheme.

Prerequisites

All the prerequisites are listed below. You can certainly use hardware and software that is not listed below, but we do not guarantee that the code will work.

Hardware

• Intel/AMD CPU with 8 performance cores or higher (recommended)
• NVIDIA GPU with compute capability 7.0 or higher (recommended)

Software

• GCC/G++ 11 or higher (recommended)
• CMake 3.20 or higher (recommended, if not, manually change the CMakeLists.txt)
• CUDA 11.0 or higher (recommended)
• IDE (optional, CLion is recommended)

Third-party libraries

We use the following third-party libraries. Please install them before building the code.

• OpenMP. Used for CPU multi-threading. Each CUDA stream is assigned to a CPU thread.

Configuring and Building

cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DCMAKE_CUDA_ARCHITECTURES=80-real
cmake --build build -j

Testing and Benchmarking

./build/bin/test_cuDilithium2
./build/bin/test_cuDilithium3
./build/bin/test_cuDilithium5

./build/bin/bench_cuDilithium2
./build/bin/bench_cuDilithium3
./build/bin/bench_cuDilithium5

Citation

If you use this code in your research, please cite the following paper:

Early access version: https://ieeexplore.ieee.org/document/10663956

@ARTICLE{10663956,
  author={Shen, Shiyu and Yang, Hao and Dai, Wangchen and Zhang, Hong and Liu, Zhe and Zhao, Yunlei},
  journal={IEEE Transactions on Parallel and Distributed Systems}, 
  title={High-Throughput GPU Implementation of Dilithium Post-Quantum Digital Signature}, 
  year={2024},
  volume={},
  number={},
  pages={1-12},
  keywords={Graphics processing units;Task analysis;Throughput;Digital signatures;Security;Streams;Parallel processing;Digital signature;GPU;parallel processing;post-quantum cryptography;Dilithium},
  doi={10.1109/TPDS.2024.3453289}}

Since the paper is in process of publication with IEEE, please refer to IACR ePrint Archive version as latest version: https://eprint.iacr.org/2024/1365

@misc{cryptoeprint:2024/1365,
      author = {Shiyu Shen and Hao Yang and Wangchen Dai and Hong Zhang and Zhe Liu and Yunlei Zhao},
      title = {High-Throughput {GPU} Implementation of Dilithium Post-Quantum Digital Signature},
      howpublished = {Cryptology {ePrint} Archive, Paper 2024/1365},
      year = {2024},
      url = {https://eprint.iacr.org/2024/1365}
}

License

This project (cuDilithium) is released under GPLv3 license. See COPYING for more information.

Some header files contain the modified code from Dilithium official repository. These codes are released under Apache 2.0 License. See Apache 2.0 License for more information.

The CUDA implementation of FIPS202 is released under MIT license. See MIT License for more information.

The FIPS202 reference implementation and the random number generator is released under CC0 license. See CC0 License for more information.