MCPcopy Create free account
hub / github.com/apache/tvm / have_cudagraph

Function have_cudagraph

python/tvm/support/nvcc.py:1103–1111  ·  view source on GitHub ↗

Either CUDA Graph support is provided

()

Source from the content-addressed store, hash-verified

1101
1102
1103def have_cudagraph():
1104 """Either CUDA Graph support is provided"""
1105 try:
1106 cuda_ver = get_cuda_version()
1107 if cuda_ver < (10, 0):
1108 return False
1109 return True
1110 except RuntimeError:
1111 return False
1112
1113
1114@tvm_ffi.register_global_func("tvm.support.nvcc.supports_bf16")

Callers

nothing calls this directly

Calls 1

get_cuda_versionFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…