additional test updates

Signed-off-by: Ben Howe <[email protected]>

Author: bmhowe23

libs/qec/python/tests/test_decoding_config.py

@@ -14,6 +14,26 @@
 
 # nv_qldpc_decoder_config tests
 
+
+def is_nv_qldpc_decoder_available():
+    """
+    Helper function to check if the NV-QLDPC decoder is available.
+    """
+    try:
+        # First check to sure a GPU is available.
+        result = subprocess.run(["nvidia-smi"], capture_output=True, timeout=2)
+        if result.returncode != 0:
+            return False
+        # Now make sure that the NV-QLDPC decoder is available.
+        H_list = [[1, 0, 0, 1, 0, 1, 1], [0, 1, 0, 1, 1, 0, 1],
+                  [0, 0, 1, 0, 1, 1, 1]]
+        H_np = np.array(H_list, dtype=np.uint8)
+        nv_dec_gpu_and_cpu = qec.get_decoder("nv-qldpc-decoder", H_np)
+        return True
+    except Exception as e:
+        return False
+
+
 FIELDS = {
     "use_sparsity": (bool, True, False),
     "error_rate": (float, 1e-3, 5e-2),
@@ -284,6 +304,9 @@ def test_configure_decoders_from_str_smoke():
     decoder_config.H_sparse = [1, 2, 3, -1, 6, 7, 8, -1, -1]
     decoder_config.set_decoder_custom_args(nv)
     yaml_str = decoder_config.to_yaml_str()
+    # Do not instantiate the decoder if it is not available.
+    if not is_nv_qldpc_decoder_available():
+        return
     status = qec.configure_decoders_from_str(yaml_str)
     assert isinstance(status, int)
     qec.finalize_decoders()