refactor: separate extension ops and backends

2026-08-16 21:15:52 +08:00
parent 3406157431
commit 6ac3b51496
18 changed files with 82 additions and 54 deletions
@@ -7,7 +7,7 @@ from typing import List, Optional
 import torch
 from torch import Tensor

-from astrai.extension.attention_backend import (
+from astrai.extension.backend.attention import (
    CudaBackend,
    get_backend,
 )