From 3089a3c767fb22ff71b0a8205413d84f10189d61 Mon Sep 17 00:00:00 2001 From: AlexandruAntonescuKeysight Date: Thu, 28 Mar 2024 03:26:39 +0200 Subject: [PATCH 1/3] Updates the github checkout action to the last version --- .github/workflows/main.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 7fd79eb8..dd54451c 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -5,7 +5,7 @@ jobs: build: runs-on: ubuntu-latest steps: - - uses: actions/checkout@v2 + - uses: actions/checkout@v4 - uses: awalsh128/cache-apt-pkgs-action@latest with: packages: make gcc g++ libprotobuf-dev protobuf-compiler From eabb2f3d0c626a98caba1dda0a53ac511be08a96 Mon Sep 17 00:00:00 2001 From: AlexandruAntonescuKeysight Date: Thu, 28 Mar 2024 05:09:38 +0200 Subject: [PATCH 2/3] simplifies modules calling to solves the the problem of redundancy --- pyproject.toml | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index 932dbd7d..6c1d45b2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,3 +29,10 @@ Repository = "https://github.com/mlcommons/chakra.git" [tool.setuptools.package-data] "chakra.et_def" = ["et_def_pb2.pyi", "et_def.proto"] + +[project.scripts] +chakra_converter = "chakra.et_converter.et_converter:main" +chakra_visualizer = "chakra.et_visualizer.et_visualizer:main" +chakra_timeline_visualizer = "chakra.et_timeline_visualizer.et_timeline_visualizer:main" +chakra_generator = "chakra.et_generator.et_generator:main" +chakra_jsonizer = "chakra.et_jsonizer.et_jsonizer:main" \ No newline at end of file From 231e8c8964af8c6a10321618caa545adf1fe17b9 Mon Sep 17 00:00:00 2001 From: AlexandruAntonescuKeysight Date: Thu, 28 Mar 2024 05:14:10 +0200 Subject: [PATCH 3/3] modifies the example in the main workflow to match the new modules calling capability --- .github/workflows/main.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index dd54451c..820bac65 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -23,10 +23,10 @@ jobs: - run: conda install pytorch - run: pip install fbgemm-gpu-cpu - run: pip install . - - run: python3 -m chakra.et_generator.et_generator --num_npus 1 + - run: chakra_generator --num_npus 1 - run: | - python3 -m chakra.et_visualizer.et_visualizer --help + chakra_visualizer --help for f in *.et do - python3 -m chakra.et_visualizer.et_visualizer --input_filename ${f} --output_filename "${f%.*}".dot + chakra_visualizer --input_filename ${f} --output_filename "${f%.*}".dot done