Skip to content

Commit

Permalink
update fft bench
Browse files Browse the repository at this point in the history
  • Loading branch information
nindanaoto committed Aug 18, 2024
1 parent 4fb591e commit 8aebfe0
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 2 deletions.
1 change: 1 addition & 0 deletions Dockerfile-ubuntu2404
Original file line number Diff line number Diff line change
Expand Up @@ -8,3 +8,4 @@ RUN DEBIAN_FRONTEND=noninteractive apt-get install -y build-essential g++ libomp
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | bash -s -- -y
ENV PATH="/root/.cargo/bin:${PATH}"
RUN wget https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB && apt-key add GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB && echo "deb https://apt.repos.intel.com/oneapi all main" | tee /etc/apt/sources.list.d/oneAPI.list && apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y intel-oneapi-mkl-devel && rm GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB
RUN rustup default nightly
17 changes: 16 additions & 1 deletion fft-bench.bash
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,12 @@ ninja
echo "Benchmarking SPQLIOS" | tee -a /tmp/log.txt
./test/nand | tee -a /tmp/log.txt

cmake . -G Ninja -B /tmp/build -DENABLE_TEST=ON -DUSE_AVX512=ON
cd /tmp/build
ninja
echo "Benchmarking SPQLIOS AVX512" | tee -a /tmp/log.txt
./test/nand | tee -a /tmp/log.txt

cd /TFHEpp
rm -rf /tmp/build
cmake . -G Ninja -B /tmp/build -DENABLE_TEST=ON -DUSE_CONCRETE_FFT=ON
Expand All @@ -16,6 +22,15 @@ ninja
echo "Benchmarking concrete-fft" | tee -a /tmp/log.txt
./test/nand | tee -a /tmp/log.txt

cd /TFHEpp
rm -rf /tmp/build
cmake . -G Ninja -B /tmp/build -DENABLE_TEST=ON -DUSE_CONCRETE_FFT=ON -DUSE_AVX512=ON
cd /tmp/build
ninja
ninja
echo "Benchmarking concrete-fft AVX512" | tee -a /tmp/log.txt
./test/nand | tee -a /tmp/log.txt

cd /TFHEpp
rm -rf /tmp/build
source /opt/intel/oneapi/setvars.sh
Expand All @@ -33,4 +48,4 @@ ninja
echo "Benchmarking FFTW3" | tee -a /tmp/log.txt
./test/nand | tee -a /tmp/log.txt

cat /tmp/log.txt
cat /tmp/log.txt
2 changes: 1 addition & 1 deletion thirdparties/concrete-fft
Submodule concrete-fft updated 1 files
+5 −1 CMakeLists.txt

0 comments on commit 8aebfe0

Please sign in to comment.