diff options
| author | Michael Feil <63565275+michaelfeil@users.noreply.github.com> | 2024-12-31 09:32:22 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-12-31 09:32:22 +0100 |
| commit | 71cd6d55337b1541f602c1afffa6baf6dd75b09c (patch) | |
| tree | 207e7d050e9c4bd685a563e457b2e9ef59b66f20 /candle-flash-attn/build.rs | |
| parent | d60eba140820326ffc7ec39a8982e91feb462732 (diff) | |
| download | candle-71cd6d55337b1541f602c1afffa6baf6dd75b09c.tar.gz candle-71cd6d55337b1541f602c1afffa6baf6dd75b09c.tar.bz2 candle-71cd6d55337b1541f602c1afffa6baf6dd75b09c.zip | |
Flash-Attn upgrade / SoftCap Candle-FlashAttn [1/n] (#2688)
* update flash-attn v1
* restore: hdim224
* add 224 flash_fwd_template
* remove whitespace
Diffstat (limited to 'candle-flash-attn/build.rs')
| -rw-r--r-- | candle-flash-attn/build.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-flash-attn/build.rs b/candle-flash-attn/build.rs index 53fec5de..37247646 100644 --- a/candle-flash-attn/build.rs +++ b/candle-flash-attn/build.rs @@ -54,6 +54,7 @@ fn main() -> Result<()> { println!("cargo:rerun-if-changed=kernels/kernel_traits.h"); println!("cargo:rerun-if-changed=kernels/block_info.h"); println!("cargo:rerun-if-changed=kernels/static_switch.h"); + println!("cargo:rerun-if-changed=kernels/hardware_info.h"); let out_dir = PathBuf::from(std::env::var("OUT_DIR").context("OUT_DIR not set")?); let build_dir = match std::env::var("CANDLE_FLASH_ATTN_BUILD_DIR") { Err(_) => |