[Feature Request] Cannot import tilelang on pure CPU machine with CUDA build #1478
Description
Required prerequisites
- I have searched the Issue Tracker that this hasn't already been reported. (comment there if it has.)
Motivation
For a training framework, most of the cases run with GPUs, but in some cases, e.g. debugging CPU dataloader, we don't want to waste GPU hours. So import tilelang with the same CUDA build should work on a machine without CUDA driver/GPU.
Now, an error may occur:
OSError: libcuda.so.1: cannot open shared object file: No such file or directory