Skip to content

Missing utils.et_generator module in chakra repository, breaking workload ET generation workflow #4

@edgelhz

Description

@edgelhz

Issue Description

When following the collectiveAPI README, the step to generate workload ET requires running:

python3 -m utils.et_generator.et_generator --num_npus 64 --num_dims 1 --default_comm_size 16384
However, this command fails with ModuleNotFoundError: No module named 'utils', and searching the chakra repository (forked from mlcommons/chakra) for et_generator returns no matches found (see screenshot).
Reproduction Steps
Clone the astra-sim/chakra repository (or check out the current commit 0e3cd40).
Navigate to the chakra directory.
Attempt to run the ET generation command:
bash
运行
python3 -m utils.et_generator.et_generator --num_npus 64 --num_dims 1 --default_comm_size 16384
Observe the ModuleNotFoundError and confirm via GitHub search that et_generator files are missing from the repository.
Expected Behavior
The utils.et_generator.et_generator module should exist in the chakra repository to support the documented workflow.
The ET generation command should run without module not found errors.
Impact
This breaks the end-to-end collective communication simulation workflow documented in collectiveAPI, as users cannot generate the required workload ET files.
Suggested Fix
Restore the utils/et_generator/ directory and its contents (including et_generator.py and necessary __init__.py files).
Or, update the documentation to reflect the current state and provide alternative methods to generate workload ET files.

Screenshot is as followed:<img width="715" height="236" alt="Image" src="https://github.com/user-attachments/assets/6475fd17-f9ba-4f5d-a8a3-42a4ba90f193" />

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions