Skip to content

Commit

Permalink
Skips ktruss docstring example for CUDA version 11.4 (#2074)
Browse files Browse the repository at this point in the history
Addresses unit test failures affecting CUDA 11.4 builds, skips ktruss doctest and avoids assertion error. Specifically addresses cugraph-gpu-matrix-branch-22.04 in ops)

Authors:
  - https://github.com/betochimas

Approvers:
  - Rick Ratzel (https://github.com/rlratzel)

URL: #2074
  • Loading branch information
betochimas authored Feb 15, 2022
1 parent c397f11 commit a6df45a
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions python/cugraph/cugraph/tests/test_doctests.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@

import cugraph
import cudf
from numba import cuda


modules_to_skip = ["dask", "proto", "raft"]
Expand Down Expand Up @@ -122,6 +123,14 @@ def test_docstring(self, docstring):
globs = dict(cudf=cudf, np=np, cugraph=cugraph, datasets_path=datasets,
scipy=scipy, pd=pd)
docstring.globs = globs

# FIXME: A 11.4 bug causes ktruss to crash in that
# environment. Skip docstring test if the cuda version is either
# 11.2 or 11.4. See ktruss_subgraph.py
if docstring.name == 'ktruss_subgraph':
if cuda.runtime.get_version() == (11, 4):
return

# Capture stdout and include failing outputs in the traceback.
doctest_stdout = io.StringIO()
with contextlib.redirect_stdout(doctest_stdout):
Expand Down

0 comments on commit a6df45a

Please sign in to comment.