Skip to content

Commit e413fde

Browse files
committed
fix torch binding
Signed-off-by: ChenxiQ <chenxi.qian.cq@outlook.com>
1 parent 38bfc20 commit e413fde

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

csrc/lightning_attention_prefill/lightning_attention_prefill_torch_adpt.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@
1414
* limitations under the License.
1515
*/
1616

17-
#ifndef LIGHTNING_ATTENTION_DECODE_TORCH_ADPT_H
18-
#define LIGHTNING_ATTENTION_DECODE_TORCH_ADPT_H
17+
#ifndef LIGHTNING_ATTENTION_PREFILL_TORCH_ADPT_H
18+
#define LIGHTNING_ATTENTION_PREFILL_TORCH_ADPT_H
1919

2020
namespace vllm_ascend {
2121
std::tuple<at::Tensor, at::Tensor> npu_lightning_attention_prefill(

0 commit comments

Comments
 (0)