Skip to content

[fix] Replace pytest.skip with requirement_not_met in conjugateGradientMultiBlockCG example (#1530) #1861

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
edenfunf wants to merge 1 commit into from
Closed

[fix] Replace pytest.skip with requirement_not_met in conjugateGradientMultiBlockCG example (#1530) #1861

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 7 additions & 8 deletions cuda_bindings/examples/4_CUDA_Libraries/conjugateGradientMultiBlockCG_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@
KernelHelper,
check_cuda_errors,
find_cuda_device,
requirement_not_met,
)

conjugate_gradient_multi_block_cg = """\
Expand Down Expand Up @@ -213,31 +214,29 @@ def gen_tridiag(i, j, val, n, nz):
def main():
tol = 1e-5

import pytest

# WAIVE: Due to bug in NVRTC
return
requirement_not_met("conjugateGradientMultiBlockCG is currently waived due to a known NVRTC issue")

if platform.system() == "Darwin":
pytest.skip("conjugateGradientMultiBlockCG is not supported on Mac OSX")
requirement_not_met("conjugateGradientMultiBlockCG is not supported on Mac OSX")

if platform.machine() == "armv7l":
pytest.skip("conjugateGradientMultiBlockCG is not supported on ARMv7")
requirement_not_met("conjugateGradientMultiBlockCG is not supported on ARMv7")

if platform.machine() == "qnx":
pytest.skip("conjugateGradientMultiBlockCG is not supported on QNX")
requirement_not_met("conjugateGradientMultiBlockCG is not supported on QNX")

# This will pick the best possible CUDA capable device
dev_id = find_cuda_device()
device_prop = check_cuda_errors(cudart.cudaGetDeviceProperties(dev_id))

if not device_prop.managedMemory:
pytest.skip("Unified Memory not supported on this device")
requirement_not_met("Unified Memory not supported on this device")

# This sample requires being run on a device that supports Cooperative Kernel
# Launch
if not device_prop.cooperativeLaunch:
pytest.skip(f"Selected GPU {dev_id} does not support Cooperative Kernel Launch")
requirement_not_met(f"Selected GPU {dev_id} does not support Cooperative Kernel Launch")

# Statistics about the GPU device
print(
Expand Down
Loading