8f54218a8ed67dc6c3df585ff89777b1e2bb2c26,pytext/optimizer/sparsifiers/tests/sparsifier_test.py,TestSparsifier,test_param_mask_with_pre_mask,#TestSparsifier#,87

Before Change

        block_sz = 3
        blockwise_sparsifier = self._get_blockwise_sparsifier(block_sz, sparsity)
        param = torch.tensor([i for i in range(100)], dtype=float).view(10, 10)
        pre_mask = torch.tensor([i % 2 for i in range(100)], dtype=float).view(10, 10)
        block_l1_norms = param.new_zeros(param.shape)
        // loop-based implementation to compute blockwise l1norm
        abs_vals = []
        for i in range(10):

After Change

                    )
                )

        nnz_unpruned = torch.nonzero(pre_mask).size(0)
        max_num_nonzeros = math.ceil(nnz_unpruned * (1 - sparsity))
        nnz_blocks = math.ceil(max_num_nonzeros / block_sz)
        abs_vals.sort(reverse=True)

Instances