Is caffe the only framework supported by MHSim? What needs to be done to support PyTorch? - Githubissues

jhxu001 / MHSim

A Simulation Framework for Memristor-based Heterogeneous Computing Architectures

1 stars 2 forks source link

Is caffe the only framework supported by MHSim? What needs to be done to support PyTorch? #2

Open saba-er opened 2 years ago

saba-er commented 2 years ago

Is Caffe the only framework supported by MHSim? What needs to be done to support PyTorch?

jhxu001 commented 2 years ago

The ZSim and the PIN tool only support the instrumentation of the C/C++ applications. We design MHSim to trace and simulate the GEMM function in these applications. Since we have re-organized the NeuroSim to support GEMM, you can replace the matrix multiplication function in PyTorch with the analog GEMM function to simulate the accuracy degrade, or count the matrix multiplications in PyTorch to calculate the latency and energy.