Skip to content

Commit 42820a2

Browse files
committed
[HPU] Add smoke unit test for hpu bridge
1 parent b1b469b commit 42820a2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

.github/workflows/_gaudi_hpu_ut.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,10 +139,12 @@ jobs:
139139
pip list | grep -E 'torch|numpy'
140140
141141
- name: Do the test
142-
continue-on-error: true
142+
working-directory: pytorch
143143
run: |
144+
pip install -r .ci/docker/requirements-ci.txt
144145
echo "Call UT test scripts"
145-
# python torch_hpu/ci/access_control_test.py
146+
export PT_HPU_LAZY_MODE=0
147+
python test/run_test.py -k test_Dropout2d_hpu --include nn/test_dropout
146148
147149
- name: Cleanup workspace
148150
if: always()

0 commit comments

Comments
 (0)