Skip to content

Commit

Permalink
docs: update figure labels; ci: correct github actions syntax
Browse files Browse the repository at this point in the history
  • Loading branch information
Eoghan O'Connell committed Nov 8, 2024
1 parent 23d367f commit b812c07
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 12 deletions.
5 changes: 2 additions & 3 deletions .github/workflows/check.yml
Original file line number Diff line number Diff line change
Expand Up @@ -39,9 +39,8 @@ jobs:
pip freeze
- name: Test with pytest
run: |
coverage run --source=qpretrieve -m pytest tests
# ignore the cupy imports, as we don't have a gpu-enabled pipeline setup
--ignore=tests/test_cupy_gpu
# ignore the cupy imports, as we don't have a gpu-enabled pipeline setup
coverage run --source=qpretrieve -m pytest tests --ignore=tests/test_cupy_gpu
- name: Lint with flake8
run: |
flake8 .
Expand Down
Binary file modified examples/fft_cupy3d_speed.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
19 changes: 10 additions & 9 deletions examples/fft_cupy3d_speed.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,11 @@
"""Fourier Transform speeds for the Cupy 3D interface
This example visualizes the normalised speed for different 3D image stacks for
the `FFTFilterCupy3D` FFT Filter
This example visualizes the speed for different batch sizes for
the `FFTFilterCupy3D` FFT Filter. The y-axis shows the speed of a single
FFT for the corresponding batch size.
- Optimum stack size for 256 images (incl. padding) is bewteen 128 and 256.
- Optimum batch size is between 64 and 256 for 256x256pix images (incl. padding).
- Here, batch size is the size of the 3D stack in z.
"""
import time
Expand Down Expand Up @@ -54,11 +56,10 @@

ax1.bar(range(len(n_transforms_list)), height=speed_norm, color='darkmagenta')
ax1.set_xticks(range(len(n_transforms_list)), labels=n_transforms_list)
ax1.set_xlabel("Number of Transforms")
ax1.set_ylabel("Speed normalised by number of transforms (s)")
ax1.set_title(f"Normalised by number of transforms")
ax1.set_xlabel("Fourier transform batch size")
ax1.set_ylabel("Speed / batch size (s)")

plt.suptitle("Speed of CuPy 3D")
plt.suptitle("Speed of FFT Interface CuPy3D")
plt.tight_layout()
# plt.show()
plt.savefig("fft_cupy3d_speed.png", dpi=150)
plt.show()
# plt.savefig("fft_cupy3d_speed.png", dpi=150)

0 comments on commit b812c07

Please sign in to comment.