Either CUDA Graph support is provided
()
| 1101 | |
| 1102 | |
| 1103 | def 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") |
nothing calls this directly
no test coverage detected
searching dependent graphs…