github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-24 11:08:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typo under extensions/ (#5330)

Browse Source
This commit is contained in:
digger yu
2024-01-30 09:55:16 +08:00
committed by GitHub
parent febed23288
commit 6a3086a505
5 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions/flash_attention/flash_attention_dao_cuda.py
View File

@@ -6,7 +6,7 @@ class FlashAttentionDaoCudaExtension(_Extension):
super().__init__(name="flash_attention_dao_cuda", support_aot=False, support_jit=False, priority=10)
def is_hardware_available(self) -> bool:
# cuda extension can only be built if cuda is availabe
# cuda extension can only be built if cuda is available
try:
import torch

2
extensions/flash_attention/flash_attention_xformers_cuda.py
View File

@@ -6,7 +6,7 @@ class FlashAttentionXformersCudaExtension(_Extension):
super().__init__(name="flash_attention_xformers_cuda", support_aot=False, support_jit=False)
def is_hardware_available(self) -> bool:
# cuda extension can only be built if cuda is availabe
# cuda extension can only be built if cuda is available
try:
import torch