Add a naive conv2d cuda kernel. (#438)

* Add a naive conv2d cuda kernel.

* Proper conv2d support on the rust side.

* Conv1d testing on gpu.

* Also use the test on gpus.

* Fix the clean-ptx target.

1 files changed, 2 insertions, 0 deletions

diff --git a/Makefile b/Makefile
index eba92821..cc0a0a36 100644
--- a/Makefile
+++ b/Makefile
@@ -2,6 +2,8 @@ clean-ptx:
 	find target -name "*.ptx" -type f -delete
 	echo "" > candle-kernels/src/lib.rs
 	touch candle-kernels/build.rs
+	touch candle-examples/build.rs
+	touch candle-flash-attn/build.rs
 
 clean:
 	cargo clean