saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Disable sparse tests in export (#132824)

Browse source 
Summary: Dynamo doesn't trace through sparse tensors in fbcode. So we should disable tests that run sparse tensors in export. We should do this to make the CI green internally.

Test Plan:
Before:
Tests finished: Pass 1409. Fail 71. Fatal 0. Skip 90. Build failure 0
After:
Tests finished: Pass 1408. Fail 0. Fatal 0. Skip 162. Build failure 0

Differential Revision: D60870543

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132824
Approved by: https://github.com/BoyuanFeng
This commit is contained in:
Tugsbayasgalan (Tugsuu) Manlaibaatar 2024-08-08 01:45:12 +00:00 committed by PyTorch MergeBot
parent fb6b001cde
commit 4ca616e6d4
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
test/export/test_sparse.py
View file

@ -7,6 +7,7 @@ import sys
import unittest
import torch
from torch._dynamo.config import is_fbcode
from torch._subclasses.fake_tensor import FakeTensor
from torch.testing._internal.common_utils import (
instantiate_parametrized_tests,
@ -79,6 +80,7 @@ class SparseActivationCSR(torch.nn.Module):
#
@unittest.skipIf(is_fbcode(), "See torch._dynamo.config")
class TestSparseProp(TestCase):
def setUp(self):
TestCase.setUp(self)