/chisel-torch

Chisel implementation of a pytorch-like interface

Primary LanguageScalaThe UnlicenseUnlicense

Chisel Torch

An Chisel implementation of a PyTorch-Like Interface for Accelerator Generation

Installation

  1. Follow instructions (https://sdkman.io) on how to install SDKMan. sdk should be in your path after installation.

  1. Install JDK and SBT
    sdk install java 22.3.r11-grl
    sdk install sbt
    To verify JDK and sbt is properly installed:
    java -version
    sdk version

  1. Install Berkeley Hardfloat from Apex's Berkeley Hardfloat Branch
    git clone git@github.com:Entropy-xcy/berkeley-hardfloat.git
    cd berkeley-hardfloat
    sbt publishLocal
    Note: during installing berkely-hardfow, a non-critical warning -Xsource is outdated might appear. Change -Xsource: 2.11 to -Xsource: 2.13 in build.sbt can get rid of the warning.

  1. Clone and Run ChiselTorch
    git clone git@github.com:Entropy-xcy/chisel-torch.git
    cd chisel-torch
    sbt "runMain chiseltorch.nn.module.SequentialBuild"

After that, Sequential.fir and Sequential.vshould be available in the current working directory.