ASTRA-sim: Example input files?

[Gaurab Das]

Hi,

We were working with ASTRA-sim for a class project, and we wanted to use different workloads from the inputs folder as was mentioned in the docs:

[Gaurab Das]

Hi,

Apologies for the double email - continued here: We were working with ASTRA-sim for a class project, and we wanted to use different workloads from the inputs folder as was mentioned in the docs:

However, the inputs folder is not given to us, and we were wondering how we could get workload files for actual ML workloads like ResNet, etc. So far, I was able to run ASTRA-sim with using the workload present in tests/rt_template folder, but I think that corresponds only to an all-gather operation. How can we get workloads for ResNet, VGG, etc?

Best,

Gaurab

---

From: Gaurab Das
Sent: Friday, November 29, 2024 4:45 PM
To: astrasi...@googlegroups.com <astrasi...@googlegroups.com>
Cc: Muhender Raj Rajvee <muhe...@mit.edu>; Anshul Agarwal <ansh...@mit.edu>
Subject: ASTRA-sim: Example input files?

 

[Eric Norige]

Gaurab,

I'm not an expert, but it looks like astra-sim workloads are currently generated via chakra; see gen.sh here:

https://github.com/astra-sim/astra-sim/tree/master/tests/rt_template/inputs/workload

If you join the mlcommons working group, you will get access to "Chakra traces/v0.0.4" via google drive that has dlrm, resnet50 and llama2 traces.

Eric Norige Chief Software Architect

---

From: astrasi...@googlegroups.com <astrasi...@googlegroups.com> on behalf of Gaurab Das <gau...@mit.edu>
Sent: Friday, November 29, 2024 1:47 PM

To: astrasi...@googlegroups.com <astrasi...@googlegroups.com>
Cc: Muhender Raj Rajvee <muhe...@mit.edu>; Anshul Agarwal <ansh...@mit.edu>

Subject: Re: ASTRA-sim: Example input files?

 

[CAUTION - External Sender]

--
You received this message because you are subscribed to the Google Groups "ASTRA-sim Users" group.
To unsubscribe from this group and stop receiving emails from it, send an email to astrasim-user...@googlegroups.com.
To view this discussion visit https://groups.google.com/d/msgid/astrasim-users/BL0PR0102MB3554F2437B3017A679B9C67BB62A2%40BL0PR0102MB3554.prod.exchangelabs.com.
For more options, visit https://groups.google.com/d/optout.