Open Haoxiang-Wang opened 4 years ago
Ah I am pretty sure the code only works on V100s or TitanV right now.
I am working on an implementation based on Jax that supports more GPU types but alas it is not out yet.
@Vaishaal Yes, V100 can indeed run the code. But I find your generate_kernel_parallel
function broke for some new dataset (MNIST-style) on a single GPU, while the single-thread generate_kernel
works fine. So I suggest you make the single-thread one as the default for the single GPU case.
Noted! What was the exception, that should not happen. I can try to fix it.
I use your docker to run
run_kernel_myrtle5.sh
on a 4x 2080ti Ubuntu 18.04 workstation, and finally encounter the following error. Could you give me a hint on how to resolve this issue?After I use Ctrl+C to interrupt it, the following error gets printed out: