gem5-pytorch

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #ai #typescript #llm #go #golang #android #machine-learning #rust #deep-learning #linux

★ 10 Open GitHub ↗

PyTorch examples modified to run in gem5 in reasonable time

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Example commands for ISCA24 tutorial

The below commands assume you are in the base codespace directory: (i.e., /workspaces/gem5-bootcamp-env).  These can be copy and pasted into a bash terminal.

Simple command to ensure PyTorch can see the GPU:

```
/usr/local/bin/gem5-vega gem5/configs/example/gpufs/mi300.py --disk-image /tmp/x86-ubuntu-gpu-ml-isca --kernel ./vmlinux-gpu-ml-isca --no-kvm-perf --app gem5-pytorch/pytorch_test.py
```

## PyTorch MNIST example command:

```
/usr/local/bin/gem5-vega gem5/configs/example/gpufs/mi300.py --disk-image /tmp/x86-ubuntu-gpu-ml-isca --kernel ./vmlinux-gpu-ml-isca --no-kvm-perf --app gem5-pytorch/MNIST/test_1batch/pytorch_qs_mnist.py
```

## Make the m5term util

```
pushd gem5/util/term ;  make ; popd
```

## Mounting disk image command:

```
mkdir mnt
mount -o loop,offset=$((2048*512)) /tmp/x86-ubuntu-gpu-ml-isca mnt

cp -r gem5-pytorch/nanoGPT/nanoGPT-ff/ mnt/root/

umount mnt
```

## nanoGPT example command:

```
/usr/local/bin/gem5-vega --debug-flags=GPUCommandProc  gem5/configs/example/gpufs/mi300.py --disk-image /tmp/x86-ubuntu-gpu-ml-isca --kernel ./vmlinux-gpu-ml-isca --no-kvm-perf --app gem5-pytorch/nanoGPT/train-ff.sh --skip-until-gpu-kernel=8 --exit-after-gpu-kernel=9
```

gem5-pytorch

Today's picks

Was this content helpful?

Recent Feedback

gem5-pytorch

Today's picks

Was this content helpful?

Recent Feedback

Download README