Skip to content

Commit a4db019

Browse files
nipung90facebook-github-bot
authored andcommitted
Call torchrec cpu tests from fbgemm test gha (pytorch#4424)
Summary: This diff kicks off the torchrec cpu unittests along with fbgemm's cpu tests when there is a pull request on fbgemm Differential Revision: D77598015
1 parent 80ed942 commit a4db019

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

.github/workflows/fbgemm_gpu_ci_cpu.yml

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,9 @@ jobs:
149149
build-target: [ "default" ]
150150
python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13" ]
151151
compiler: [ "gcc", "clang" ]
152+
permissions:
153+
id-token: write
154+
contents: read
152155
needs: build_artifact
153156

154157
steps:
@@ -195,8 +198,20 @@ jobs:
195198
timeout-minutes: ${{ matrix.host-machine.timeout }}
196199
run: . $PRELUDE; test_all_fbgemm_gpu_modules $BUILD_ENV
197200

201+
- name: Test with Torchrec CPU tests
202+
uses: pytorch/torchrec/.github/workflows/reusable_unittest_ci_cpu.yml@nipun/make_cpp_tests_reusable
203+
198204
- name: Push Wheel to PyPI
199205
if: ${{ (github.event_name == 'schedule' || (github.event_name == 'workflow_dispatch' && github.event.inputs.publish_to_pypi == 'true')) && matrix.compiler == 'gcc' }}
200206
env:
201207
PYPI_TOKEN: ${{ secrets.PYPI_TOKEN }}
202208
run: . $PRELUDE; publish_to_pypi $BUILD_ENV "$PYPI_TOKEN" *.whl
209+
210+
211+
# Run torchrec CPU tests
212+
# torchrec_cpu_tests:
213+
# needs: test_and_publish_artifact
214+
# permissions:
215+
# id-token: write
216+
# contents: read
217+
# uses: pytorch/torchrec/.github/workflows/reusable_unittest_ci_cpu.yml@nipun/make_cpp_tests_reusable

0 commit comments

Comments
 (0)