diff options
Diffstat (limited to 'candle-flash-attn/src/lib.rs')
| -rw-r--r-- | candle-flash-attn/src/lib.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-flash-attn/src/lib.rs b/candle-flash-attn/src/lib.rs index 99b05229..092743f1 100644 --- a/candle-flash-attn/src/lib.rs +++ b/candle-flash-attn/src/lib.rs @@ -17,7 +17,7 @@ fn round_multiple(x: usize, m: usize) -> usize { impl candle::CustomOp3 for FlashAttn { fn name(&self) -> &'static str { - "flash-hdim32-sm80" + "flash-attn" } fn cpu_fwd( @@ -192,7 +192,7 @@ struct FlashAttnVarLen { impl candle::CustomOp3 for FlashAttnVarLen { fn name(&self) -> &'static str { - "flash-hdim32-sm80" + "flash-attn-varlen" } fn cpu_fwd( |