Skip to content
Testing PyTorch 2.4

Remove CUDA synchronizations by slicing input tensor with int instead of CUDA tensors in nn.LinearEmbeddingEncoder #2612

Sign in to view logs

Remove CUDA synchronizations by slicing input tensor with int instead of CUDA tensors in nn.LinearEmbeddingEncoder

Remove CUDA synchronizations by slicing input tensor with int instead of CUDA tensors in nn.LinearEmbeddingEncoder #2612

Run time

Learn about OS pricing on GitHub Actions
Job Run time
pytest
1m 58s
1m 58s