frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[dynamo] Eagerly install guards (#111415)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/111415
Approved by: https://github.com/voznesenskym
ghstack dependencies: #111306
This commit is contained in:
Jason Ansel
2023-11-07 08:12:57 -08:00
committed by PyTorch MergeBot
parent 2964682490
commit 9664190952
30 changed files with 333 additions and 622 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_pruning_op.py
View File

@ -4,7 +4,7 @@ import hypothesis.strategies as st
from hypothesis import given
import numpy as np
import torch
from torch.testing._internal.common_utils import TestCase, run_tests
from torch.testing._internal.common_utils import TestCase, run_tests, skipIfTorchDynamo
import torch.testing._internal.hypothesis_utils as hu
hu.assert_deadline_disabled()
@ -56,6 +56,7 @@ class PruningOpTest(TestCase):
self.assertEqual(pt_compressed_indices_map.dtype, indices_type)
@skipIfTorchDynamo()
@given(
embedding_rows=st.integers(1, 100),
embedding_dims=st.integers(1, 100),
@ -67,6 +68,7 @@ class PruningOpTest(TestCase):
self._test_rowwise_prune_op(embedding_rows, embedding_dims, torch.int, weights_dtype)
@skipIfTorchDynamo()
@given(
embedding_rows=st.integers(1, 100),
embedding_dims=st.integers(1, 100),