Returns a dictionary of compile-time info — minimal Python fallback. The native ``support.GetLibInfo`` global function is no longer registered after the upstream sync, so we synthesize the values from build-time hints instead.
()
| 20 | |
| 21 | |
| 22 | def libinfo(): |
| 23 | """Returns a dictionary of compile-time info — minimal Python fallback. |
| 24 | |
| 25 | The native ``support.GetLibInfo`` global function is no longer registered |
| 26 | after the upstream sync, so we synthesize the values from build-time hints |
| 27 | instead. |
| 28 | """ |
| 29 | return { |
| 30 | "USE_CUDA": os.environ.get("TVM_USE_CUDA", "ON"), |
| 31 | "USE_LLVM": os.environ.get("TVM_USE_LLVM", "ON"), |
| 32 | "USE_NCCL": os.environ.get("TVM_USE_NCCL", "ON"), |
| 33 | "USE_NVTX": os.environ.get("TVM_USE_NVTX", "ON"), |
| 34 | "USE_NVSHMEM": os.environ.get("TVM_USE_NVSHMEM", "OFF"), |
| 35 | "USE_HEXAGON": "OFF", |
| 36 | "USE_CUDNN": "OFF", |
| 37 | "USE_CUTLASS": "OFF", |
| 38 | "USE_VULKAN": "OFF", |
| 39 | "USE_OPENCL": "OFF", |
| 40 | "USE_METAL": "OFF", |
| 41 | "USE_ROCM": "OFF", |
| 42 | "USE_CLML": "OFF", |
| 43 | "USE_NNAPI_RUNTIME": "OFF", |
| 44 | "USE_NNAPI_CODEGEN": "OFF", |
| 45 | } |