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.
2025-06-15 18:28:24 +00:00 · 2023-08-14 10:34:42 +01:00
parent eab54e4490
commit 34f4b3187e
4 changed files with 151 additions and 30 deletions
--- a/2
+++ b/2
@ -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