def test_assumed_runtime_version():
# When CUDA Python is enabled, CuPy calculates the CUDA runtime version
# from NVRTC version. This test ensures that the assumption is correct
# by running the same logic in non-CUDA Python environment.
# When this fails, `runtime.runtimeGetVersion()` logic needs to be fixed.
(major, minor) = nvrtc.getVersion()
assert runtime.runtimeGetVersion() == major * 1000 + minor * 10
def _get_nvrtc_version():
global _nvrtc_version
if _nvrtc_version is None:
_nvrtc_version = nvrtc.getVersion()
return _nvrtc_version
