Remove CUDA synchronizations by slicing input tensor with int instead of CUDA tensors in nn.LinearEmbeddingEncoder
#2612
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Testing PyTorch 2.4 | |
| on: # yamllint disable-line rule:truthy | |
| push: | |
| branches: | |
| - master | |
| pull_request: | |
| jobs: | |
| pytest: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| with: | |
| fetch-depth: 40 | |
| # Skip workflow if only certain files have been changed. | |
| - name: Get changed files | |
| id: changed-files-specific | |
| uses: tj-actions/changed-files@v41 | |
| with: | |
| files: | | |
| docs/** | |
| examples/** | |
| README.md | |
| CHANGELOG.md | |
| - name: Setup packages | |
| if: steps.changed-files-specific.outputs.only_changed != 'true' | |
| uses: ./.github/actions/setup | |
| - name: Install main package | |
| if: steps.changed-files-specific.outputs.only_changed != 'true' | |
| run: | | |
| pip install -e .[full,test] | |
| - name: Run tests | |
| if: steps.changed-files-specific.outputs.only_changed != 'true' | |
| run: | | |
| pytest --cov --cov-report=xml | |
| - name: Upload coverage | |
| if: steps.changed-files-specific.outputs.only_changed != 'true' | |
| uses: codecov/codecov-action@v2 | |
| with: | |
| fail_ci_if_error: false |