This repository is used for HBM2-PIM lectures at the KAIST AI-PIM Center. The lab lectures consist of slides explaining the lab and simulator code from this repository.
This simulator repository is based on the the HBM2-PIM simulator repository. The existing HBM2-PIM simulator is managed and maintained at https://github.com/SAITPublic/PIMSimulator. Please refer to this link to use it.
Scons
tool for compiling PIMSimulator:
sudo apt install scons
gtest
for running test cases:
sudo apt install libgtest-dev
- To Install PIMSimulator:
# compile
scons
- Show a list of test cases
./sim --gtest_list_tests
# Example
PIMKernelFixture.
gemv_tree
gemv
mul
add
relu
MemBandwidthFixture.
hbm_read_bandwidth
hbm_write_bandwidth
PIMBenchFixture.
gemv
mul
add
relu
- Test Running
# Running: functionality test (GEMV)
./sim --gtest_filter=PIMKernelFixture.gemv
# Running: functionality test (MUL)
./sim --gtest_filter=PIMKernelFixture.mul
# Running: performance test (GEMV)
./sim --gtest_filter=PIMBenchFixture.gemv
# Running: performance test (ADD)
./sim --gtest_filter=PIMBenchFixture.add
More information about this simulator can be found here