[Draft] Add Ascend NPU optim Support

.github/workflows/build-wheel.yml

@@ -0,0 +1,67 @@
+name: Build Wheel
+
+on:
+  push:
+    branches:
+      - main
+    paths-ignore:
+      - 'docs/**'
+      - '.assets/**'
+      - '**.md'
+      - '.gitignore'
+      - '.gitattributes'
+      - 'LICENSE'
+  pull_request:
+    branches:
+      - main
+    paths-ignore:
+      - 'docs/**'
+      - '.assets/**'
+      - '**.md'
+      - '.gitignore'
+      - '.gitattributes'
+      - 'LICENSE'
+  workflow_dispatch:
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.12"]
+
+    steps:
+      - uses: actions/checkout@v4 
+        with:
+          fetch-depth: 0
+          fetch-tags: true
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          cache: 'pip'
+
+      - name: Install build
+        run: |
+          pip install -U pip && pip install build
+
+      - name: Check build configuration
+        run: |
+          if [ ! -f "pyproject.toml" ]; then
+            echo "Error: pyproject.toml not found! Cannot build wheel."
+            exit 1
+          fi
+          echo "✅ Build configuration file (pyproject.toml) found."
+
+      - name: Build Wheel
+        run: |
+          python -m build --wheel --outdir dist
+          ls -la dist/
+
+      - name: Upload Wheel Artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: cache-dist-wheel-py${{ matrix.python-version }}
+          path: dist/*.whl
+          retention-days: 15

.github/workflows/check-mkdocs.yml

@@ -0,0 +1,70 @@
+name: Check MkDocs Build
+
+on:
+  push:
+    branches:
+      - main
+    paths-ignore:
+      - '.github/**'
+      - '.gitignore'
+      - '.gitattributes'
+      - 'LICENSE'
+  pull_request:
+    branches:
+      - main
+    paths-ignore:
+      - '.github/**'
+      - '.gitignore'
+      - '.gitattributes'
+      - 'LICENSE'
+  workflow_dispatch:
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.12"]
+
+    steps:
+      - uses: actions/checkout@v4 
+        with:
+          fetch-depth: 0
+          fetch-tags: true
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          cache: 'pip'
+
+      - name: Check build configuration
+        run: |
+          if [ ! -f "pyproject.toml" ]; then
+            echo "Error: pyproject.toml not found! Cannot build docs."
+            exit 1
+          fi
+          echo "✅ Build configuration file (pyproject.toml) found."
+
+      - name: Install Only MkDocs Dependencies
+        run: |
+          pip install mkdocs>=1.5.0 mkdocs-api-autonav mkdocs-material \
+              mkdocstrings-python mkdocs-gen-files mkdocs-awesome-nav \
+              mkdocs-glightbox mkdocs-git-revision-date-localized-plugin \
+              mkdocs-minify-plugin regex ruff pydantic
+
+      # Build MkDocs documentation with strict mode and fail it if the 
+      # logs contains 'WARNING', 'aborted', 'ERROR', or not contains 
+      # 'Documentation built'.
+      - name: Build MkDocs Documentation Strictly
+        run: |
+          mkdocs build --strict 2>&1 | tee build.log
+          if grep -E 'WARNING|aborted|abort|ERROR' build.log; then
+            echo "MkDocs build failed due to warnings or errors."
+            exit 1
+          elif ! grep -q 'Documentation built' build.log; then
+            echo "MkDocs build did not complete successfully."
+            exit 1
+          else
+            echo "MkDocs build completed successfully!"
+          fi

.github/workflows/cpu-tests.yml

@@ -0,0 +1,105 @@
+name: Run CPU Tests
+
+on:
+  push:
+    branches:
+      - main
+    paths-ignore:
+      - 'docs/**'
+      - '.assets/**'
+      - '**.md'
+      - '.gitignore'
+      - '.gitattributes'
+      - 'LICENSE'
+  pull_request:
+    branches:
+      - main
+    paths-ignore:
+      - 'docs/**'
+      - '.assets/**'
+      - '**.md'
+      - '.gitignore'
+      - '.gitattributes'
+      - 'LICENSE'
+  workflow_run:
+    workflows: ["Build Wheel"] 
+    branches: [main]          
+    types: [completed] 
+  workflow_dispatch:           
+
+jobs:
+  Basic_CPU_Tests:
+    if: >-
+      (github.event_name == 'workflow_run' && github.event.workflow_run.conclusion == 'success') ||
+      github.event_name == 'pull_request' ||
+      github.event_name == 'push' ||
+      github.event_name == 'workflow_dispatch'
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.12"]
+
+    steps:
+      - name: Checkout code (for test files)
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          cache: 'pip'
+
+      - name: Download built wheel artifact (only for workflow_run)
+        if: github.event_name == 'workflow_run'
+        uses: dawidd6/action-download-artifact@v6 
+        with:
+          workflow: build_wheel.yml              
+          run_id: ${{ github.event.workflow_run.id }} 
+          name: cache-dist-wheel-py${{ matrix.python-version }}
+          path: dist/                           
+
+      - name: Install dependencies
+        run: |
+          pip install -U pip && pip install torch==2.9.1 torchvision --index-url https://download.pytorch.org/whl/cpu  
+          if [ -f "dist/*.whl" ]; then
+            pip install dist/*.whl
+          else
+            pip install -e .
+          fi
+          pip install pytest
+
+      - name: Run Forward Pattern CPU Tests (pytest)
+        run: |
+          cd tests
+          pytest api/test_forward_pattern.py -v -s -x
+
+      - name: Run Refresh Context CPU Tests (pytest)
+        run: |
+          cd tests
+          pytest api/test_refresh_context.py -v -s -x
+
+      - name: Run TaylorSeers CPU Tests (pytest)
+        run: |
+          cd tests
+          pytest api/test_taylorseers.py -v -s -x
+
+      - name: Run Load Configs CPU Tests (pytest)
+        run: |
+          cd tests
+          pytest api/test_load_configs.py -v -s -x
+
+      - name: Verify cache_dit installation
+        run: |
+          python -c "import cache_dit; print(f'✅ cache_dit imported successfully, version: {cache_dit.__version__}')"
+        continue-on-error: false
+
+      - name: Upload test logs (optional)
+        uses: actions/upload-artifact@v4
+        with:
+          name: test-logs-py${{ matrix.python-version }}
+          path: tests/basic-cpu-tests-results.txt
+          retention-days: 7
+        if: always()

.github/workflows/pre-commit.yml

@@ -0,0 +1,26 @@
+name: pre-commit
+
+on:
+  pull_request:
+    branches: [main]
+  push:
+    branches: [main]
+  workflow_dispatch: # for manual trigger
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: ${{ github.event_name == 'pull_request' }}
+
+permissions:
+  contents: read
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+          cache: 'pip'
+      - uses: pre-commit/action@v3.0.1

.pre-commit-config.yaml

@@ -5,7 +5,7 @@ repos:
       - id: check-docstring-first
       - id: check-toml
       - id: check-yaml
-        exclude: packaging/.*
+        exclude: ^(packaging/.*|mkdocs\.yml)$
         args:
           - --allow-multiple-documents
       - id: mixed-line-ending

.readthedocs.yml

@@ -0,0 +1,24 @@
+# Read the Docs configuration file
+# See https://docs.readthedocs.io/en/stable/config-file/v2.html for details
+
+version: 2
+
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.12"
+  jobs:
+    post_checkout:
+      - git fetch --unshallow || true
+
+mkdocs:
+  configuration: mkdocs.yml
+  fail_on_warning: true
+
+# Optionally declare the Python requirements required to build your docs
+python:
+  install:
+    - method: pip
+      path: .
+      extra_requirements:
+        - docs

LICENSE

@@ -1,4 +1,4 @@
-Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved
+Copyright (c) 2025 Cache-DiT Authors. All Rights Reserved
 
                                  Apache License
                            Version 2.0, January 2004