Refactor repository's structure and add build CI

Author: Victor-Jung

.github/workflows/CI.yml

@@ -0,0 +1,24 @@
+name: CI
+
+on: 
+  push:
+  pull_request:
+  workflow_dispatch:
+
+jobs:
+
+  build-deepquant:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout Repo
+        uses: actions/checkout@v4
+        with:
+          submodules: recursive
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+            python-version: "3.11"
+      - name: Build DeepQuant
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e .

.gitignore

@@ -28,3 +28,4 @@ dist/
 *.onnx
 *.npz
 onnx/*
+Dataset/*

src/DeepQuant/custom_forwards/__init__.py DeepQuant/custom_forwards/__init__.pysrc/DeepQuant/custom_forwards/__init__.py renamed to DeepQuant/custom_forwards/__init__.py

@@ -68,22 +68,12 @@ def exportBrevitas(
         nn.Module: An FX GraphModule with explicit quantization operations.
     """
 
-    # EXPORT_FOLDER = (
-    #     Path().cwd()
-    # )  # Initialize export folder to current working directory
-    # print(EXPORT_FOLDER)  # Display export folder path for reference
-    # if Path().cwd().name != "onnx":  # Check if already in 'onnx' directory
-    #     EXPORT_FOLDER = (
-    #         EXPORT_FOLDER / "onnx"
-    #     )  # If not, create/use an 'onnx' subdirectory
-
     EXPORT_FOLDER = Path().cwd()
-    print(EXPORT_FOLDER)
-    if Path().cwd().name != "onnx":
-        EXPORT_FOLDER = EXPORT_FOLDER / "onnx"
+    if Path().cwd().name == "DeepQuant":
+        EXPORT_FOLDER = EXPORT_FOLDER / "Tests/ONNX"
         EXPORT_FOLDER.mkdir(parents=True, exist_ok=True)
 
-    printer = GraphModulePrinter()  # Instantiation of the custom printer
+    printer = GraphModulePrinter()
 
     ###############################################################################
     # 1. Original Network
@@ -105,12 +95,6 @@ def exportBrevitas(
             example_input
         )  # Compute original model output on example input for validation
 
-    # torch.onnx.export(  
-    #     model,  # Model to export
-    #     args=example_input,  # Example input for tracing
-    #     f=EXPORT_FOLDER / "1_model_qcdq_original.onnx",
-    #     opset_version=13,
-    # )
     export_onnx_qcdq(  # Export original model to ONNX format with QCDQ (Quant-Cast-DeQuant) nodes
         model,  # Model to export
         args=example_input,  # Example input for tracing