Develop

.github/workflows/docker-image.yml

@@ -6,7 +6,7 @@ on:
 
 jobs:
   build-and-test:
-    runs-on: ubuntu-latest
+    runs-on: self-hosted
 
     permissions:
       contents: read
@@ -35,6 +35,7 @@ jobs:
           context: .
           file: ./Dockerfile
           push: true
+          no-cache: true
           tags: ghcr.io/psal-postech/torchsim-test:${{ github.sha }}
 
       # Step 4: Wait for GHCR propagation

.github/workflows/pytorchsim_test.yml

@@ -662,3 +662,37 @@ jobs:
             -e TORCHSIM_VECTOR_LANE="${{ inputs.vector_lane }}" \
             -e TORCHSIM_SPAD_SIZE="${{ inputs.spad_size }}" \
             ${{ inputs.image_name }} python3 PyTorchSim/tests/test_scheduler.py
+
+  test_accuracy:
+    name: Run test_accuracy
+    runs-on: self-hosted
+    steps:
+      - name: Log in to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Prepare volume directory
+        run: mkdir -p /tmp/torchsim-ci/${GITHUB_SHA}
+
+      - name: Run run_cycle.sh
+        run: |
+          echo "Running run_cycle.sh"
+          docker run --rm \
+            -v /tmp/torchsim-ci/${GITHUB_SHA}:/dump \
+            -e TORCHSIM_DUMP_PATH=/dump \
+            -e TORCHSIM_VECTOR_LANE="${{ inputs.vector_lane }}" \
+            -e TORCHSIM_SPAD_SIZE="${{ inputs.spad_size }}" \
+            ${{ inputs.image_name }} bash -c \
+            "cd /workspace && PyTorchSim/experiments/artifact/cycle_validation/run_cycle.sh && \
+            cp PyTorchSim/experiments/artifact/cycle_validation/summary_cycle.out /dump/summary_cycle.out"
+          ls /tmp/torchsim-ci/${GITHUB_SHA}
+
+      - name: Upload Accuracy Report Artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: accuracy-report
+          path: /tmp/torchsim-ci/${{ github.sha }}/summary_cycle.out
+          if-no-files-found: error

.gitignore

@@ -1,3 +1,3 @@
 __pycache__/
-PyTorchSimBackend/build/
+TOGSim/build/
 .vscode

.gitmodules

@@ -1,18 +1,15 @@
-[submodule "PyTorchSimBackend/extern/onnx"]
-	path = PyTorchSimBackend/extern/onnx
+[submodule "TOGSim/extern/onnx"]
+	path = TOGSim/extern/onnx
 	url = https://github.com/onnx/onnx.git
-[submodule "PyTorchSimBackend/extern/protobuf"]
-	path = PyTorchSimBackend/extern/protobuf
+[submodule "TOGSim/extern/protobuf"]
+	path = TOGSim/extern/protobuf
 	url = https://github.com/protocolbuffers/protobuf.git
-[submodule "PyTorchSimBackend/extern/booksim"]
-	path = PyTorchSimBackend/extern/booksim
+[submodule "TOGSim/extern/booksim"]
+	path = TOGSim/extern/booksim
 	url = https://github.com/PSAL-POSTECH/booksim.git
-[submodule "PyTorchSimBackend/extern/torch2timeloop"]
-        path = PyTorchSimBackend/extern/torch2timeloop
-        url = https://github.com/Accelergy-Project/pytorch2timeloop-converter.git
-[submodule "PyTorchSimBackend/extern/ramulator2"]
-	path = PyTorchSimBackend/extern/ramulator2
+[submodule "TOGSim/extern/ramulator2"]
+	path = TOGSim/extern/ramulator2
 	url = https://github.com/PSAL-POSTECH/ramulator2
-[submodule "PyTorchSimBackend/extern/stonneCore"]
-	path = PyTorchSimBackend/extern/stonneCore
+[submodule "TOGSim/extern/stonneCore"]
+	path = TOGSim/extern/stonneCore
 	url = https://github.com/PSAL-POSTECH/stonne_core.git

Dockerfile

@@ -4,7 +4,7 @@ FROM ghcr.io/psal-postech/torchsim_base:latest
 # Prepare PyTorchSim project
 COPY . /workspace/PyTorchSim
 
-RUN cd PyTorchSim/PyTorchSimBackend && \
+RUN cd PyTorchSim/TOGSim && \
     mkdir -p build && \
     cd build && \
     conan install .. --build=missing && \

Dockerfile.ksc2025

@@ -0,0 +1,90 @@
+# Copyright (c) 2020 The Regents of the University of California
+# All Rights Reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met: redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer;
+# redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in the
+# documentation and/or other materials provided with the distribution;
+# neither the name of the copyright holders nor the names of its
+# contributors may be used to endorse or promote products derived from
+# this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
+
+# Copied from Gem5 Docker file
+ENV DEBIAN_FRONTEND=noninteractive
+RUN apt -y update && apt -y upgrade && \
+    apt -y install build-essential git m4 scons zlib1g zlib1g-dev \
+    libprotobuf-dev protobuf-compiler libprotoc-dev libgoogle-perftools-dev \
+    python3-dev python-is-python3 doxygen libboost-all-dev \
+    libhdf5-serial-dev python3-pydot libpng-dev libelf-dev pkg-config pip \
+    python3-venv black libssl-dev libasan5 libubsan1
+RUN pip install mypy pre-commit jupyter
+
+# Pass Access Token securely
+ENV PATH=$PATH:/root/.local/bin
+ENV LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:/opt/conda/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:$LD_LIBRARY_PATH
+
+# Build Gem5
+RUN git clone https://github.com/PSAL-POSTECH/gem5.git --branch TorchSim
+RUN cd gem5 && scons build/RISCV/gem5.opt -j $(nproc)
+ENV GEM5_PATH=/workspace/gem5/build/RISCV/gem5.opt
+
+# Build LLVM RISC-V
+RUN git clone https://github.com/PSAL-POSTECH/llvm-project.git --branch torchsim --depth 1
+RUN cd llvm-project && mkdir build && cd build && \
+    cmake -DLLVM_ENABLE_PROJECTS=mlir -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=/riscv-llvm -DLLVM_TARGETS_TO_BUILD=RISCV -G "Unix Makefiles" ../llvm && \
+    make -j && make install
+
+# Store RISC-V LLVM for TorchSim
+ENV TORCHSIM_LLVM_PATH=/riscv-llvm/bin
+ENV TORCHSIM_LLVM_INCLUDE_PATH=/riscv-llvm/include
+ENV TORCHSIM_DIR=/workspace/PyTorchSim
+ENV LLVM_DIR=/riscv-llvm
+
+# Download RISC-V tool chain
+RUN apt install -y wget && \
+    wget https://github.com/riscv-collab/riscv-gnu-toolchain/releases/download/2023.12.14/riscv64-glibc-ubuntu-22.04-llvm-nightly-2023.12.14-nightly.tar.gz && \
+    wget https://github.com/riscv-collab/riscv-gnu-toolchain/releases/download/2023.12.14/riscv64-elf-ubuntu-20.04-llvm-nightly-2023.12.14-nightly.tar.gz && \
+    tar -zxvf riscv64-elf-ubuntu-20.04-llvm-nightly-2023.12.14-nightly.tar.gz && tar -zxvf riscv64-elf-ubuntu-20.04-llvm-nightly-2023.12.14-nightly.tar.gz && \
+    rm *.tar.gz
+
+ENV RISCV=/workspace/riscv
+ENV PATH=$RISCV/bin:$PATH
+
+# Install Spike simulator
+RUN apt -y install device-tree-compiler
+RUN git clone https://github.com/PSAL-POSTECH/riscv-isa-sim.git --branch TorchSim && cd riscv-isa-sim && mkdir build && cd build && \
+    ../configure --prefix=$RISCV && make -j && make install
+
+# Install Proxy kernel
+RUN git clone https://github.com/riscv-software-src/riscv-pk.git && \
+     cd riscv-pk && git checkout 4f3debe4d04f56d31089c1c716a27e2d5245e9a1 && mkdir build && cd build && \
+    ../configure --prefix=$RISCV --host=riscv64-unknown-elf && make -j && make install
+
+# Install torchsim dependency
+RUN apt install ninja-build && pip install onnx matplotlib && pip install --user conan==1.56.0
+
+# Prepare ONNXim project
+RUN git clone https://github.com/PSAL-POSTECH/PyTorchSim.git --branch tutorial
+RUN cd PyTorchSim/TOGSim && \
+    git submodule update --recursive --init && \
+    mkdir -p build && \
+    cd build && \
+    conan install .. --build=missing && \
+    cmake .. && \
+    make -j$(nproc)