You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
def test_random_small():
it = 10
for i in range(it):
Nc = random.randint(3, 10)
X = torch.randn(A_N, Nc)
X_c = X.to(gpu)
print('X_c shape', X_c.shape)
print(X_c)
AX_d = A_d @ X
assert(torch.allclose(AX_d, A@X))
E AssertionError: assert False
E + where False = <built-in method allclose of type object at 0x7facd8b4f680>(tensor([[-8.0905e-01, -1.2791e+00, 2.1114e+00, -2.1343e+00, -5.8658e+00],\n [ 2.3513e+00, 1.4423e+00, -2.9569e...00, -1.0391e+00, -5.9623e-01, -4.8469e+00],\n [-2.3855e+00, -1.8825e+00, -4.6459e+00, -8.6935e-01, 3.7740e+00]]), (<16x16 sparse matrix tensor of type 'torch.float32'\n with 46 stored elements in Compressed Sparse Row format> @ tensor([[ 2.3843e-01, -1.6153e-01, 5.9662e-01, -1.7112e+00, -2.1326e+00],\n [ 1.2859e+00, 9.5606e-01, -9.1817e...02, -1.1889e+00, 5.0320e-02, -1.0626e+00],\n [-6.7612e-01, -9.1202e-01, -2.9174e+00, -4.0951e-01, 1.3557e+00]])))
E + where <built-in method allclose of type object at 0x7facd8b4f680> = torch.allclose
After fixing build issues with #12, two tests failed
root@51e1485d94b7:/TorchVision/numml# pytest numml/tests
============================================================= test session starts ==============================================================
platform linux -- Python 3.10.12, pytest-7.4.2, pluggy-1.3.0
rootdir: /TorchVision/numml
plugins: shard-0.1.2, hypothesis-5.35.1, rerunfailures-12.0, xdist-3.3.1, xdoctest-1.0.2, flakefinder-1.1.0
collected 33 items
Running 33 items in this shard
numml/tests/test_add.py ... [ 9%]
numml/tests/test_misc.py . [ 12%]
numml/tests/test_spdmm.py .FF.. [ 27%]
numml/tests/test_spmm.py ....... [ 48%]
numml/tests/test_spmv.py ...... [ 66%]
numml/tests/test_sptrsv.py ...... [ 84%]
numml/tests/test_transpose.py ..... [100%]
=================================================================== FAILURES ===================================================================
______________________________________________________________ test_random_small _______________________________________________________________
E AssertionError: assert False
E + where False = <built-in method allclose of type object at 0x7facd8b4f680>(tensor([[-8.0905e-01, -1.2791e+00, 2.1114e+00, -2.1343e+00, -5.8658e+00],\n [ 2.3513e+00, 1.4423e+00, -2.9569e...00, -1.0391e+00, -5.9623e-01, -4.8469e+00],\n [-2.3855e+00, -1.8825e+00, -4.6459e+00, -8.6935e-01, 3.7740e+00]]), (<16x16 sparse matrix tensor of type 'torch.float32'\n with 46 stored elements in Compressed Sparse Row format> @ tensor([[ 2.3843e-01, -1.6153e-01, 5.9662e-01, -1.7112e+00, -2.1326e+00],\n [ 1.2859e+00, 9.5606e-01, -9.1817e...02, -1.1889e+00, 5.0320e-02, -1.0626e+00],\n [-6.7612e-01, -9.1202e-01, -2.9174e+00, -4.0951e-01, 1.3557e+00]])))
E + where <built-in method allclose of type object at 0x7facd8b4f680> = torch.allclose
numml/tests/test_spdmm.py:43: AssertionError
------------------------------------------------------------- Captured stdout call -------------------------------------------------------------
X_c shape torch.Size([16, 7])
tensor([[-1.4404, 0.9939, -0.2333, -2.4501, -2.3983, -1.0288, -1.5292],
[-1.1281, -1.9627, -1.0358, 1.0002, 1.2289, 0.0903, 0.7665],
[-0.8329, 1.7655, -1.1728, -0.1793, -1.4545, 0.0141, -1.2997],
[-1.7013, -0.9780, 0.5817, -1.3643, 1.5334, 0.6137, 0.5878],
[ 0.4688, 0.1218, 0.8664, -1.0398, -0.2259, 2.2077, -0.3796],
[-1.3176, -0.0595, 2.0575, 1.5325, -0.7838, -0.1799, 0.7953],
[ 0.2436, -2.7121, 0.0796, -0.2203, 0.2659, 0.9575, -0.4840],
[ 0.5368, 1.4393, -0.4030, -0.5211, -0.5826, -1.3434, -1.5291],
[-0.0376, 0.3784, 1.1600, 1.3821, 1.3911, -1.4542, -1.5568],
[ 1.1818, 0.9652, 1.0323, 0.5053, -0.0812, -1.3966, 0.6781],
[-1.1266, -0.4934, 1.1943, 0.5738, -1.2757, 0.7131, 1.4122],
[-0.6217, -0.2607, -1.1545, -0.5109, 1.4249, -0.0774, -1.4336],
[ 0.0060, -1.3877, 0.3932, -0.8707, -0.5830, 0.7535, 0.9527],
[ 0.1212, -0.0844, -0.8639, 0.1214, 0.6723, 1.7152, -0.2713],
[-1.3067, 0.8132, 0.4800, -2.3894, 0.9714, 0.4336, -1.1133],
[ 0.7836, 0.4225, 0.0109, -1.2518, -1.2735, -0.3628, -0.5556]],
device='cuda:0')
X_c shape torch.Size([16, 6])
tensor([[-1.6990, 0.4994, 0.5237, 0.5681, -0.4384, 1.2278],
[-0.5838, 0.1167, 0.7284, 0.8329, 0.0827, -0.0480],
[ 0.2050, 1.3393, -1.5127, -0.9296, -1.1537, 1.8951],
[-0.7593, -0.6198, -1.6280, -1.8682, -1.8920, 0.1182],
[ 1.7719, -0.4381, -1.1646, -1.5857, -2.2401, 1.6927],
[ 1.2371, -0.7035, -0.5878, -0.5940, 1.0570, -0.1962],
[-1.1086, -2.3059, 0.3850, -0.6142, 1.1206, 0.3247],
[ 0.2323, -0.7529, -0.6670, -0.0404, 1.4693, -0.9065],
[-1.6904, -0.4596, -0.3784, 2.1711, 0.8126, 2.7665],
[-1.9220, 0.6336, -0.8892, -0.5160, 1.7726, 0.6870],
[-1.9859, 0.8231, -1.2517, -0.6015, 0.6485, 0.5330],
[-0.7359, -0.4978, -0.1557, 0.4194, -2.9327, -1.5208],
[-0.6135, -0.6400, -1.1061, -1.6915, -0.3025, 0.9853],
[-1.4310, 1.7253, -0.6249, -0.2569, 1.1259, 1.0723],
[ 0.5355, -0.5477, -0.3024, -0.2601, 0.0102, -0.1565],
[ 0.0215, -0.2547, -0.2166, -0.5851, 0.4905, 0.5024]],
device='cuda:0')
X_c shape torch.Size([16, 4])
tensor([[ 0.1297, -0.6206, -1.3438, -0.2378],
[ 0.3542, -2.1727, -0.7693, 0.1832],
[-2.0531, -0.2708, 0.4542, -0.7819],
[-0.6752, -0.1414, 0.8351, -0.5958],
[ 0.4637, -0.6149, 1.0920, -0.4849],
[ 0.0196, 0.5081, -1.0010, -0.2156],
[ 0.3359, 0.7699, -2.2112, -1.8252],
[ 0.6903, 1.0862, -1.0192, 1.3107],
[-0.2100, -1.2446, -0.4514, 0.6777],
[ 0.8924, -1.0165, 0.4438, -0.0431],
[ 1.4194, -0.1714, -0.0895, 0.0612],
[ 0.1829, 0.6200, 0.8062, -0.8499],
[-0.5414, -1.6655, 0.6044, 0.9376],
[ 0.2957, 0.7887, 0.4509, 0.1071],
[ 0.0645, -0.5269, -1.9421, -1.1922],
[ 1.1843, -1.8400, -0.0630, -0.2972]], device='cuda:0')
X_c shape torch.Size([16, 7])
tensor([[ 1.4878e+00, -4.0182e-01, 2.2026e+00, -5.4703e-01, -1.5376e+00,
-1.0124e+00, -1.7798e-01],
[ 3.1039e+00, 1.3340e+00, -6.4154e-01, 2.7985e+00, -1.0064e+00,
-1.2514e-01, -8.9587e-01],
[ 6.9080e-01, -5.0479e-03, -5.4363e-01, -1.3511e+00, 2.3356e-01,
-4.4866e-01, -2.3332e+00],
[ 3.6111e-01, -5.2257e-01, -8.4437e-01, 1.4156e+00, -2.1662e-01,
-6.6325e-01, -1.4181e-01],
[ 7.6379e-01, -1.5159e+00, 2.3365e-01, 2.8765e-01, 1.8937e+00,
-1.7489e-01, -5.9196e-01],
[ 6.6749e-01, 1.1846e+00, 5.3885e-01, 1.5514e+00, 1.6079e+00,
1.3231e+00, -6.3911e-01],
[-5.4540e-02, 4.1947e-01, -1.6453e-01, 2.9159e+00, -8.3123e-01,
-1.0161e+00, 7.0982e-01],
[-4.6030e-01, 1.4809e+00, 1.4565e+00, -1.4879e+00, -1.6303e-02,
1.0789e+00, 4.5324e-01],
[-6.6465e-01, -5.4234e-01, -8.2486e-01, 4.7613e-01, -4.2572e-01,
3.7492e-01, 1.8957e-01],
[ 2.1584e-01, -1.6270e+00, -4.7006e-02, -2.4820e+00, -3.4601e-01,
-1.2760e-01, -4.9856e-01],
[-1.7065e+00, 4.8403e-01, 5.1602e-01, -9.9702e-01, 7.8479e-01,
1.3271e+00, -3.4873e-01],
[ 1.2887e+00, -7.6163e-01, -5.9705e-01, -2.8159e-02, -1.8242e-01,
1.0442e+00, -3.6987e-01],
[-3.2847e-01, 7.2676e-01, -1.1154e+00, -2.7572e-01, 1.0034e+00,
-3.6583e-01, -6.5653e-01],
[ 1.1268e+00, -2.1012e-01, 9.4497e-01, 1.4431e-03, 1.4943e-01,
-2.2855e+00, 1.1309e+00],
[ 8.3730e-01, 8.8567e-01, 2.4612e+00, 2.3620e-01, 3.0178e-01,
3.9953e-01, 7.6512e-01],
[ 1.2079e+00, -7.4040e-01, -1.3897e-01, 1.8495e+00, -1.0368e+00,
6.5823e-01, 1.1544e+00]], device='cuda:0')
X_c shape torch.Size([16, 9])
tensor([[ 1.1608e+00, 5.2404e-01, -9.3447e-01, 8.4869e-01, -1.6773e-01,
7.4402e-01, 3.3067e-01, 1.3007e+00, -9.4865e-02],
[ 5.6259e-01, 5.4271e-01, 1.9070e+00, 9.6501e-01, -2.9819e-01,
1.2864e+00, 1.5435e+00, -6.1125e-01, -5.1652e-01],
[-1.0947e+00, 3.3532e-02, 1.6763e+00, -1.9187e-01, 8.5340e-01,
-2.9716e-01, -6.9661e-01, -1.4154e+00, 4.9596e-01],
[-1.5290e+00, 4.0677e-01, 1.1593e+00, 6.7598e-01, 3.6662e-01,
-7.1687e-01, -5.2561e-01, 9.8510e-01, -1.2586e+00],
[-6.7048e-01, 3.8925e-01, 3.8757e-01, 3.9510e-01, -2.3837e-01,
-3.9225e-01, -2.2230e-01, 3.3597e-01, 7.5240e-01],
[-7.9070e-02, 4.8587e-01, -3.1904e-01, -1.8768e+00, -1.2998e-01,
-1.8312e+00, 8.6905e-01, 1.4286e+00, 4.3833e-01],
[-7.1978e-01, 4.6438e-01, 2.9762e+00, 6.1793e-01, 5.5364e-01,
-7.6758e-01, -3.9675e-02, 2.0812e+00, -6.5035e-01],
[ 1.4078e+00, 5.3098e-01, 1.7682e+00, 1.2678e+00, 2.0861e-01,
2.8300e+00, -4.1974e-01, -1.3530e+00, -2.2785e+00],
[-1.8561e+00, -1.2427e-01, 9.4302e-01, 2.0212e+00, 5.6076e-01,
-8.7159e-01, -2.2218e-01, -8.7846e-01, 9.1332e-01],
[ 6.2230e-02, -2.7110e+00, -9.0537e-01, -7.0086e-01, -4.0505e-01,
-1.0166e+00, 1.7999e+00, 9.1718e-01, 1.1454e+00],
[ 1.9940e+00, -1.5937e+00, -1.3100e+00, -1.5429e-01, 1.4527e+00,
-2.9646e-01, -6.5418e-01, -1.5161e+00, -5.7280e-01],
[-2.1334e-01, -2.4639e+00, 2.1572e+00, 8.8547e-01, -9.6068e-01,
-7.8268e-01, -3.3942e-01, -1.7174e-01, -2.7654e-01],
[ 1.2832e-01, -5.5500e-01, 1.1688e-01, 7.3154e-01, -1.5545e+00,
-6.0215e-01, -1.7040e-01, -4.1701e-02, 5.9348e-01],
[ 1.0569e+00, -6.8000e-01, -1.0715e+00, -1.2657e+00, -6.0673e-01,
7.7856e-01, -2.0067e-03, 2.1625e-01, -3.8065e-01],
[ 4.5311e-01, 7.3426e-01, -3.1287e+00, -4.6920e-01, -6.8173e-01,
-5.4592e-01, -2.2556e+00, -1.1776e+00, -2.0990e-01],
[ 6.1662e-01, 1.3936e+00, -7.9541e-01, 1.5003e+00, -3.3568e-01,
5.5757e-01, -6.6173e-01, 1.5195e+00, -1.0046e+00]], device='cuda:0')
X_c shape torch.Size([16, 10])
tensor([[ 1.2644, 0.9153, -1.4336, 0.7042, 0.6132, 0.2158, 0.2887, 0.5955,
-1.4903, 0.4799],
[ 0.4747, -0.5038, 0.2039, 0.4999, 1.4249, -0.6385, -0.1321, 0.0411,
-0.0987, 0.5058],
[-0.2237, 1.0426, 1.3616, 2.1435, -0.2582, -0.1537, 0.6851, -0.1747,
2.4723, 0.8146],
[ 1.1464, 0.1538, 0.2178, -2.1835, -0.5254, 0.5341, 0.0711, 1.0723,
1.5233, 0.8579],
[-0.2565, -0.6761, 1.3049, -1.2890, -0.9296, -0.3167, -1.4590, -0.3813,
0.6864, 0.9067],
[-0.0045, -0.6211, -1.3761, -1.8368, -0.7799, -1.4577, 1.2145, -0.4969,
0.8128, -0.0879],
[-0.2819, 1.1318, 0.5851, 1.5346, -0.5031, 1.0811, 1.4598, -0.2922,
-1.0698, -0.6239],
[ 1.5007, 0.4776, 0.5756, -0.1036, 0.3009, -1.4226, 0.1117, 0.0685,
1.4614, -0.6837],
[ 0.2659, -0.1216, 0.4524, -0.8880, -0.0903, 0.0514, -0.3307, 1.3727,
0.4569, 0.1330],
[-1.3085, 0.3482, 0.8168, 1.4885, -1.0427, 0.9636, -0.2865, -0.1337,
0.4671, -0.9516],
[ 0.3257, 0.4804, 0.8782, -2.1633, 2.6182, -1.2561, 1.5588, 0.4776,
-1.4705, -1.6446],
[ 0.0566, 0.8201, -0.8358, 1.4604, -0.6825, 0.1172, -0.0529, -1.3245,
1.3474, 0.5735],
[ 0.8320, -0.2725, -0.8190, 0.2139, -0.6147, 1.8202, 0.8190, -0.1298,
0.7880, 0.4006],
[-0.2897, 0.3501, -0.0573, -1.8452, -1.0914, 0.7466, 1.7042, 0.8599,
0.2791, 0.3213],
[-0.8833, 1.0887, -1.6795, 0.2387, 1.3289, 1.0220, -0.4792, -0.3981,
-0.2759, -0.3885],
[ 0.3195, 1.4533, -0.5149, -0.4989, 0.0384, 0.1927, 1.1188, -1.7946,
0.1168, 0.0319]], device='cuda:0')
X_c shape torch.Size([16, 9])
tensor([[-0.1261, 1.1235, -1.0696, -1.1387, 0.2743, -0.5753, 0.5699, 0.5149,
-1.5009],
[-0.0163, 0.8872, 0.9338, 0.0028, -0.7828, -1.7374, 0.0482, 0.6348,
-0.6524],
[-2.3185, -1.5195, 1.6479, 0.4001, 1.1418, 1.3328, 0.3003, 1.3518,
0.4094],
[-1.4756, -1.3459, -0.3559, 2.3344, 0.4912, -0.2207, -0.5194, 1.3649,
-0.1092],
[ 2.0059, -0.7026, -1.1630, 0.3063, -0.3568, -0.4022, -0.1154, -2.0210,
-0.2269],
[ 0.1390, 0.1577, -0.4332, 1.3558, 1.3769, 0.7643, 0.7959, 0.1422,
0.6731],
[-1.1134, -1.7257, -0.6762, 0.2199, -0.7438, -1.0302, -1.3823, 1.9751,
-0.2361],
[-2.2063, 0.1420, -2.1714, 0.4063, 0.1646, 0.2098, 0.7066, -0.1464,
-0.3691],
[-0.2976, -2.1344, 0.8700, 0.5639, 1.2127, -1.7652, -1.5514, 1.2440,
-0.0477],
[ 1.1270, -2.5411, -2.1104, 0.6838, 0.5407, 0.3054, 0.0783, -0.6014,
-1.3387],
[-0.9373, 1.9748, -0.3357, 1.2090, -1.0583, -0.2297, -0.8720, 0.2073,
0.0692],
[ 0.4263, -0.9635, 0.3399, -0.1582, 1.5743, -1.2042, -1.2970, 1.5115,
-0.8326],
[-1.3463, -1.0039, -1.3930, 1.9752, 0.1632, -1.1564, -1.1395, 1.0007,
-0.0174],
[-0.7713, 1.0199, -0.1383, 0.6839, -1.4275, -1.4894, 0.5242, 1.1992,
-1.1920],
[ 1.3805, 0.6728, -0.1094, 1.4066, 0.8237, 0.3745, -1.4426, 2.3637,
-1.2894],
[ 0.7094, -0.9870, -0.8011, -1.8140, -0.1459, -0.4534, -1.2952, -0.4278,
0.9144]], device='cuda:0')
X_c shape torch.Size([16, 5])
tensor([[ 2.3843e-01, -1.6153e-01, 5.9662e-01, -1.7112e+00, -2.1326e+00],
[ 1.2859e+00, 9.5606e-01, -9.1817e-01, -1.2881e+00, 1.6006e+00],
[-1.7942e-02, 6.3140e-01, 5.2398e-01, -2.7630e-01, 1.5381e+00],
[-5.1545e-01, 6.3432e-01, -6.3058e-01, 1.4679e-02, 1.4887e+00],
[-1.0812e+00, -5.2346e-02, 2.2321e-01, 8.6249e-02, 7.4195e-01],
[ 3.1881e-01, 5.8811e-01, 1.0562e+00, 1.3342e-01, 2.9426e-01],
[-1.1309e+00, 2.4535e+00, -5.7642e-01, 2.8675e-01, -1.9409e-01],
[-1.0309e+00, -5.4439e-01, -7.7303e-01, 3.2667e-02, 1.8129e+00],
[-6.7964e-04, -8.9906e-01, 2.0343e+00, 3.6000e-01, -2.7563e-01],
[-3.0619e-01, 1.5841e+00, 1.5015e+00, -8.4094e-01, 3.2306e-01],
[-2.0362e-01, -5.3436e-01, 9.7085e-01, 9.9474e-01, -3.6796e-01],
[-8.1160e-01, 2.9902e+00, 1.3559e-01, -5.7836e-01, -7.6623e-01],
[ 1.2366e+00, 8.4008e-01, 3.7793e-01, 4.3765e-01, -3.7793e-01],
[-9.6175e-01, -2.6040e-01, 1.5787e+00, 1.1064e+00, 1.3660e+00],
[ 1.0333e+00, 5.8460e-02, -1.1889e+00, 5.0320e-02, -1.0626e+00],
[-6.7612e-01, -9.1202e-01, -2.9174e+00, -4.0951e-01, 1.3557e+00]],
device='cuda:0')
______________________________________________________________ test_random_large _______________________________________________________________
E AssertionError: assert False
E + where False = <built-in method allclose of type object at 0x7facd8b4f680>(tensor([[-3.1022, -0.1757, 0.2265, 2.6405, 2.6282],\n [ 6.8254, 1.6969, -2.2275, -5.1129, -1.2788],\n ...4.5011],\n [-0.8021, -0.4414, 1.2198, -0.4238, -4.7221],\n [ 3.6725, -0.1307, -2.1412, -0.1562, 3.5673]]), (<2048x2048 sparse matrix tensor of type 'torch.float32'\n with 6142 stored elements in Compressed Sparse Row format> @ tensor([[-0.4234, 0.3763, -0.2115, -0.0504, 1.1204],\n [ 2.2554, 0.9283, -0.6494, -2.7413, -0.3875],\n ...1.1468],\n [ 1.0123, -0.3853, 0.2250, -0.2176, -1.1945],\n [ 2.3424, -0.2580, -0.9581, -0.1869, 1.1864]])))
E + where <built-in method allclose of type object at 0x7facd8b4f680> = torch.allclose
numml/tests/test_spdmm.py:55: AssertionError
=========================================================== short test summary info ============================================================
FAILED numml/tests/test_spdmm.py::test_random_small - AssertionError: assert False
FAILED numml/tests/test_spdmm.py::test_random_large - AssertionError: assert False
======================================================== 2 failed, 31 passed in 12.38s =========================================================
root@51e1485d94b7:/TorchVision/numml#
The text was updated successfully, but these errors were encountered: