pytorch - 💡(How to fix) Fix DISABLED test_variant_consistency_jit_grid_sampler_2d_cuda_float32 (__main__.TestJitCUDA) [1 comments, 1 participants]

Official PRs (…)
ON THIS PAGE

Recommended Tools

×6

Utilities matched from this issue’s tags and category — try them while you read without losing context.

GitHub issue graph ai analysis

Paste a GitHub issue URL. We fetch that issue, discover linked issues from bodies/comments/timeline, collect linked pull requests, and produce a structured English report.

The report is written in English Markdown for sharing and archival.

Helpful · Quick feedback

Loading…
GitHub stats
pytorch/pytorch#179743Fetched 2026-04-09 07:50:13
View on GitHub
Comments
1
Participants
1
Timeline
4
Reactions
0
Participants
Timeline (top)
labeled ×3commented ×1

Error Message

Traceback (most recent call last): File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 109, in test_variant_consistency_jit self.indiv_variant_test_jit( File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 157, in indiv_variant_test_jit check_against_reference( File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_jit.py", line 140, in check_against_reference self.assertEqual(g2, g2_test, atol=5e-4, rtol=1e-4) File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 4440, in assertEqual raise error_metas.pop()[0].to_error( # type: ignore[index] AssertionError: Tensor-likes are not close!

Mismatched elements: 84 / 520 (16.2%) Greatest absolute difference: 0.005322886630892754 at index (0, 4, 8, 0) (up to 0.0005 allowed) Greatest relative difference: 11.676852226257324 at index (0, 8, 9, 0) (up to 0.0001 allowed)

The above exception was the direct cause of the following exception:

Traceback (most recent call last): File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 1175, in test_wrapper return test(*args, **kwargs) File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 120, in test_variant_consistency_jit raise Exception(variant_error_info) from e # noqa: TRY002 Exception: Error testing grid_sampler_2d function variant with dtype: torch.float32 with inputs SampleInput(input=tensor([[[[-0.7560, -0.1443, -1.3317, 1.0707, -0.0536, 1.2746, -1.0104, 1.7234, 0.4455, 0.2850, 0.4191, 1.4160, -1.6165, 1.2780, -1.4350, 0.3450, -1.6000, 0.3590, 1.1890, 1.1320], [-1.2507, -0.5309, -1.4537, 0.7853, 1.7886, -0.0907, 1.6612, 1.2864, 0.1156, -0.8906, -0.7386, -1.8012, -1.8938, 0.8986, 0.0282, -0.9127, 0.3131, -1.2557, 0.7019, 1.8979], [-0.9708, 1.1602, -1.4717, -1.8265, -1.3409, 1.7040, 0.1004, -0.4273, 0.3805, 0.4131, -1.8698, -0.5582, -1.2652, 0.8626, 1.7328, 1.7129, -1.0489, 0.3309, -0.0435, 1.1554], [ 1.2676, -1.9798, -0.7505, -0.2731, -1.1362, 0.5353, -1.3947, 1.6763, -1.4323, 0.5500, 0.3345, -0.1932, 0.2958, -1.0673, 0.1891, 1.8858, 1.9893, -0.4553, 0.4284, -0.4182], [ 0.8169, -1.6031, -1.4634, 0.5036, -0.9926, 1.9319, -1.1357, 1.4299, -0.1737, -0.3567, -0.5193, -0.1565, 1.3256, 0.9212, -1.0795, 0.2933, -0.1425, -0.9356, -0.3506, -1.1891]],

[[-1.4523, 1.6652, 1.8988, -1.5228, -0.4680, 1.7696, -0.0126, 0.0595, -0.9792, 1.3731, -1.0529, 1.3009, -1.9354, -0.4431, -0.3360, -0.8617, 1.6389, 0.5681, 1.4091, -0.4532], [ 0.4200, 1.6801, -0.6350, -0.2937, 0.8192, -1.7547, -0.0023, -0.8332, 1.7256, 1.5671, -1.9655, -0.9337, 1.1646, 0.1157, 0.8171, -0.2420, 1.9833, -1.2515, -1.9892, 1.2424], [ 1.4888, -0.1588, 1.3372, -1.7930, -0.2623, 0.0159, 1.6651, -1.4205, -1.6741, 0.3484, -0.7193, 1.5819, 0.8232, 1.1013, -1.7786, -0.8769, 0.7951, -1.1791, 0.6205, -0.4372], [ 1.8010, -0.6551, -1.6349, -0.1433, -1.6172, 1.7753, -0.3452, -1.8068, 0.2773, 1.0525, 0.2656, -0.0981, -0.6501, -1.4674, 1.1974, 0.1889, -1.1462, 1.2498, 0.0458, -1.5549], [-0.7784, -1.4418, 1.5518, -1.6349, 0.2633, -0.0242, 0.4860, -0.2963, 1.9413, 0.8316, -1.9919, -1.5023, -0.7553, 1.7810, -0.1282, 1.8717, -0.6003, 1.1921, -1.3241, -0.8015]],

[[ 0.0387, -0.0683, -1.3119, -0.6655, 1.3270, 0.4466, -0.5344, 0.1092, -1.6954, -1.7021, 0.2210, 0.0358, 1.7904, -1.5685, -0.0929, -0.3973, 0.2937, 0.8790, 0.6636, 1.0969], [ 1.1867, -1.8680, -0.7894, 0.1756, -1.8204, 0.7961, -1.3391, 0.7083, -1.5092, 1.1195, -1.0322, -0.1478, -0.9416, -1.6585, 1.1229, 0.8414, 1.6447, 0.3158, 1.7880, 0.9119], [ 0.8099, 0.3150, 0.6882, 1.6934, -0.2961, 0.3199, 0.9885, -0.3853, 1.1251, -0.8356, 1.4918, 1.3761, 1.7323, -1.2555, -1.9041, 0.2660, -1.5036, -1.1210, 1.7914, 0.6311], [ 1.9929, 1.1131, 1.9899, 1.0328, -1.4152, 1.5587, -0.4122, -1.5055, 1.6023, -1.2365, -0.9438, 1.2561, -1.5768, 0.6131, 1.9208, 1.6280, -1.4219, 1.7101, 0.0306, 0.4187], [-1.7767, 1.2516, 0.7330, -1.1223, 0.2094, 0.5165, 1.4007, 0.6846, -0.1133, 0.6371, 0.6199, -0.0956, 1.5327, -1.6353, 0.7567, 0.6291, -1.2143, -1.4736, 0.0165, 0.8720]]],

[[[-1.3789, -1.2527, 1.3401, 0.3406, -0.7042, 1.1389, 0.6666, -0.4117, -0.2989, -0.3938, 0.3820, -0.6539, 0.9630, 0.9078, -1.2130, 0.3959, 1.1905, -0.8307, 0.9276, 1.8631], [ 0.6441, -0.0742, -0.2414, 1.8217, 0.9493, 0.8255, -0.7547, -1.2031, -1.3122, 0.5966, 0.7229, -1.6750, 0.0700, -0.6009, 0.2622, 1.1515, -1.0787, -1.4436, -1.4295, 0.3572], [ 0.6079, -0.5907, -0.9739, -0.1204, -1.8401, -0.1114, 0.1417, 1.0373, -0.0542, 1.5353, 0.0925, -1.4203, -0.6187, -0.3825, 1.4880, 0.5443, 1.3337, -1.5342, 0.1816, 0.3702], [-1.8681, 0.5987, -1.0082, -1.9412, -0.0964, -1.7483, 1.1809, 0.5175, 1.9990, 1.1500, -1.7001, 0.5029, -0.1091, -1.1460, 1.7729, -0.5552, -1.4418, 1.4706, -1.9570, 0.4494], [ 1.4411, 0.6376, -0.3329, -0.5400, -0.1568, 0.9053, -0.7919, -0.5293, 0.5903, 0.0579, 1.8277, -1.6401, -0.8383, 0.5593, 0.4174, -0.9361, -0.5755, -0.3629, 1.9672, 1.2225]],

[[-0.5613, -0.3929, 0.2896, -1.7571, 1.9991, -0.4219, -0.7767, -1.8055, -0.3581, -1.7616, -1.6126, 0.2505, -0.2944, 0.4794, -0.3436, 0.2436, 1.0697, -0.7204, 0.8678, 1.1480], [ 1.6177, 0.9862, 0.8313, -1.5676, -0.8029, -0.4710, -0.9152, 1.3049, 1.1056, 1.7808, 0.6238, -1.4056, 1.3417, 0.1611, 0.7123, -1.9622, -1.0206, 1.0915, 1.0982, 1.9980], [-0.3426, 1.1918, 1.3012, -1.3486, -1.1785, 0.6574, 0.4834, -1.5202, -1.4326, -1.0696, 0.7712, 0.6510, 1.4800, 0.4209, -1.5200, -1.4426, -0.4412, -0.0307, 0.9065, -1.8860], [-0.7202, 0.5170, -0.7852, -0.3009, -0.4691, 1.9814, 1.9398, -1.0492, 1.9136, -0.2763, -0.0766, -0.8461, -0.1436, 1.9201, 1.0925, 0.1738, 0.6860, 0.4168, 0.5323, 0.8805], [-0.4933, -1.5355, -1.1386, -1.8716, -0.6593, -0.4551, -1.8895, -0.0941, 0.6320, -1.5665, 0.2567, -1.6547, 0.4524, 0.8572, 0.4672, 0.3772, 0.9701, -0.6117, -0.4405, 0.2073]],

[[ 0.8025, -0.0595, -0.9438, 0.2717, -0.3834, 0.8637, 0.9916, -0.4114, 1.8228, -0.6555, -0.1733, -0.6234, 0.4483, 0.4614, 0.2109, -0.0100, -0.7163, -0.0803, -0.5773, 1.8567], [ 1.1786, 0.0330, 0.9112, 0.1013, -0.1249, -0.5918, -1.8807, -0.6012, -0.4238, 1.9679, -1.3820, 1.6903, -1.8332, -0.8515, 0.0218, 0.7700, -1.7483, 0.7853, -1.9223, 0.1178], [-0.6113, -1.2863, -1.8943, 0.6953, -1.7618, 0.8581, 0.9284, 1.7700, -0.1035, -1.8848, 0.2061, -1.4119, -0.3077, 0.3535, -1.3055, 1.3411, 0.9784, 0.2072, 1.7066, -1.2979], [-1.7509, 1.1533, 1.9451, -0.6432, -1.2304, -1.1662, -1.2911, -0.8734, 0.9899, -1.9781, -0.4201, 1.5957, -1.5980, -0.7309, -1.9116, -0.0167, -1.6116, -1.2348, 0.5920, 1.1657], [-0.1331, 1.6117, 0.5441, -1.8298, -1.6769, 1.5804, -1.6245, 1.8640, -1.0085, 1.5367, 0.4123, 0.5132, -0.4380, 1.5536, -1.8067, 1.9436, -0.7693, -0.0835, 1.0801, 1.2580]]]], device='cuda:0', requires_grad=True), args=(tensor([[[[ 0.8840, 0.7284], [-1.9067, -1.7008], [ 1.8565, -0.3989], [-0.8446, 0.7945], [-1.1669, 0.7130], [-1.8342, -1.8440], [-0.0883, 1.5642], [ 0.7254, 0.5880], [ 0.2307, 1.1799], [ 1.4357, -0.6544]],

[[-1.0316, 1.2807], [ 1.0162, -0.9042], [ 0.5910, 0.3717], [ 0.3884, 0.6074], [-1.0467, -1.9530], [-0.7783, -0.1390], [ 1.5114, -1.0019], [-1.1427, 0.9776], [-0.7653, -0.7106], [ 0.5124, 0.0094]],

[[-0.2061, -1.0028], [ 0.5493, 0.3959], [-0.0231, 0.9015], [-0.1169, 1.0216], [ 1.6647, -0.0388], [-1.0319, 0.4510], [ 1.5703, 0.4492], [-1.9579, -0.3020], [ 0.7635, 1.5684], [ 0.3821, 1.6603]],

[[-0.6688, 0.5106], [-0.4162, 1.6228], [-0.3385, 1.9604], [ 0.7010, -1.9517], [-1.5050, 0.1919], [-1.9553, -1.1527], [-0.1650, 0.0051], [-0.8922, 1.1805], [-0.0818, 1.2613], [ 0.7639, -1.4357]],

[[-1.6299, 0.4255], [-0.6200, -0.6759], [ 1.0267, -1.6362], [ 1.2146, -1.7742], [-1.9981, -1.0791], [-0.7641, 1.6715], [-1.6763, -1.4935], [ 1.0545, 1.9053], [-1.8099, 0.2377], [ 1.8880, -1.7312]],

[[ 1.1423, -1.4683], [ 1.4219, 0.7878], [-0.3497, -0.4911], [ 0.3866, 0.7407], [-1.4957, 0.8446], [ 0.3847, -1.4483], [-0.3230, 0.6764], [-0.1395, -1.7608], [ 0.4753, -0.1717], [-1.9776, -1.1859]],

[[ 0.9461, 1.6932], [ 1.0734, -0.9379], [-0.2483, -1.6817], [ 0.3628, -0.4302], [ 0.8477, -0.1925], [ 1.5683, -1.4538], [-0.2494, -1.6042], [ 0.0142, -1.6684], [-1.6479, -0.3535], [ 1.6242, -1.8027]],

[[-1.2505, -1.3212], [ 1.9157, -0.6191], [-1.7512, 0.5165], [-1.5732, -0.1249], [-0.5898, -0.4158], [-0.4769, -0.3535], [ 0.4879, -0.2919], [-0.1908, 1.6932], [ 1.0552, 1.9789], [ 0.4114, -1.8693]],

[[-0.6473, -0.2875], [ 0.8754, 0.5485], [ 1.7395, 0.5469], [ 1.9870, -0.5567], [-1.5295, 0.1025], [-1.3862, 1.2063], [-0.7479, -0.0594], [ 1.6344, -0.8996], [ 0.5685, -0.6133], [-1.8893, 1.9618]],

[[-1.3969, -1.2584], [-0.5230, 0.7359], [-1.9951, 1.8692], [-1.7383, 0.5308], [ 0.0171, 1.5884], [-0.4782, 1.9707], [ 0.3105, 1.5377], [-0.3061, 1.2858], [-0.0608, 0.7897], [-1.5873, 1.1970]],

[[ 1.3557, -1.8997], [ 0.0401, 1.0429], [ 0.1588, 0.3099], [ 0.8134, -1.7725], [-1.4868, -1.9436], [-1.3589, -1.8519], [ 0.0235, -0.0903], [-0.9190, -0.7169], [-0.4357, -0.9308], [ 1.2138, -0.4648]],

[[ 0.7425, -1.2723], [-1.2882, -0.9645], [ 0.9654, -0.3355], [ 1.1874, 1.9311], [-1.2859, -0.2779], [ 0.0146, 0.6645], [-1.9226, -1.8894], [-1.9399, 0.3529], [-0.9138, 1.9704], [-1.1756, 0.3996]],

[[-0.0967, -1.5186], [-0.5572, 1.4003], [ 1.1635, 0.3844], [-1.8407, 1.2133], [ 0.3440, 0.8430], [ 1.6675, -0.2081], [ 1.5182, -0.0991], [ 0.4312, 1.1988], [ 1.0257, 0.9181], [-1.3318, 0.0113]]],

[[[ 0.5229, 1.7186], [ 0.2447, -1.7722], [ 1.9558, 0.7897], [-0.2866, 1.7106], [-1.8656, -0.6073], [ 0.2541, -0.5758], [ 0.7862, -0.7967], [ 0.2434, -0.2500], [ 1.9724, -1.6783], [-1.2378, -1.8009]],

[[-0.2075, 1.0700], [-1.8012, -0.7954], [ 0.7769, -0.3403], [ 1.7522, -1.6351], [ 1.1443, -0.0109], [ 1.1499, 1.1863], [ 0.5536, -1.2255], [ 1.1178, -1.0296], [ 0.5785, -1.0306], [ 0.4561, 0.3083]],

[[ 1.8243, -0.4109], [-0.8581, -0.9415], [ 1.6291, 1.4080], [ 0.2663, -1.4106], [-1.3930, 0.4565], [ 1.3747, -1.4464], [-0.5173, 0.6275], [-1.4881, -0.0791], [-1.5300, -0.0376], [-0.7357, 1.3118]],

[[ 0.4213, -0.4424], [ 0.4718, 0.8544], [ 0.6342, 0.5930], [-1.4592, 0.9840], [-0.2018, -1.5982], [ 0.2531, 0.1849], [ 0.7382, -1.9418], [-0.8996, 0.3930], [ 0.3761, 0.0904], [-1.3371, 0.9920]],

[[ 0.8061, -0.4647], [-1.5188, 1.0175], [ 1.3355, -0.9962], [-1.2889, -1.5765], [-0.2926, 1.8030], [-1.2162, 1.3498], [-1.0174, -0.0172], [ 1.9736, -0.9195], [-0.5556, -0.3043], [-0.8305, 0.3111]],

[[-0.1680, 1.6404], [ 0.2090, 0.4003], [ 1.7559, -0.1290], [-1.2025, 1.6680], [-1.0993, 1.4024], [-1.6462, -1.7576], [ 0.3758, -0.5544], [ 1.7620, 0.0187], [ 0.7033, 1.0926], [ 1.8603, 1.8435]],

[[-0.2956, -0.1107], [-0.9312, 0.3183], [-0.9891, -1.5630], [-1.4147, -0.0128], [-1.5586, 1.3915], [ 1.1812, -1.4658], [-1.8646, 0.9345], [-0.0903, -0.1244], [ 0.4643, -0.7193], [ 0.2759, -0.7039]],

[[ 1.6475, -0.4765], [ 1.7192, 0.9523], [-0.3220, -0.6531], [ 0.3992, 1.8709], [-1.9119, 1.7273], [ 0.7452, 1.1067], [-0.1234, 0.8440], [ 1.7273, -0.4334], [ 0.4186, -0.4803], [-1.5332, -0.1467]],

[[ 1.7229, 0.9690], [-1.8442, -1.0074], [ 0.4095, 0.1997], [ 1.2701, -1.8232], [-1.7041, -0.3767], [-0.7734, -0.2795], [-1.3943, -0.6950], [ 0.2118, 1.2398], [-1.0847, -0.2509], [-0.7644, 1.8773]],

[[ 0.3199, -1.6492], [ 0.4723, -0.6576], [ 0.3100, -1.8890], [ 1.3837, -0.1127], [ 0.4850, -1.1013], [-1.5918, -0.1211], [ 0.1576, -0.7470], [-0.6888, -1.3469], [-0.9145, 0.3586], [-1.7643, 0.4087]],

[[-0.7426, 0.7931], [-1.1826, -1.7763], [ 0.4773, -1.5527], [ 1.6368, -0.7878], [ 1.8296, -0.6230], [ 1.6000, -0.3928], [-0.2718, -1.6651], [ 1.1806, 0.5856], [ 1.9963, -0.4580], [-0.0518, -0.0662]],

[[-1.5275, -0.2008], [-1.0342, -1.4198], [ 1.5619, -0.8835], [ 0.4056, 0.8474], [ 0.6425, 1.1953], [ 1.0211, -1.4575], [ 1.4458, 0.1762], [-0.9242, -1.8079], [-1.2674, -0.0075], [-1.2202, -1.8033]],

[[ 0.8573, 0.8268], [ 1.0164, 1.8607], [-0.3526, -0.6622], [-0.1924, 0.5389], [ 1.6591, 1.8509], [ 1.1368, 0.9931], [-0.0709, 1.3218], [ 1.5008, 0.9046], [-1.4040, 1.6321], [ 1.0302, -0.2132]]]], device='cuda:0', requires_grad=True), 2, 1, False), kwargs={}, broadcasts_input=False, name=''):

The above exception was the direct cause of the following exception:

Traceback (most recent call last): File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 3443, in wrapper method(*args, **kwargs) File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 3443, in wrapper method(*args, **kwargs) File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 439, in instantiated_test result = test(self, **param_kwargs) File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 1820, in wrapper fn(*args, **kwargs) File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 1187, in test_wrapper raise e_tracked from e Exception: Error testing grid_sampler_2d function variant with dtype: torch.float32 with inputs SampleInput(input=tensor([[[[-0.7560, -0.1443, -1.3317, 1.0707, -0.0536, 1.2746, -1.0104, 1.7234, 0.4455, 0.2850, 0.4191, 1.4160, -1.6165, 1.2780, -1.4350, 0.3450, -1.6000, 0.3590, 1.1890, 1.1320], [-1.2507, -0.5309, -1.4537, 0.7853, 1.7886, -0.0907, 1.6612, 1.2864, 0.1156, -0.8906, -0.7386, -1.8012, -1.8938, 0.8986, 0.0282, -0.9127, 0.3131, -1.2557, 0.7019, 1.8979], [-0.9708, 1.1602, -1.4717, -1.8265, -1.3409, 1.7040, 0.1004, -0.4273, 0.3805, 0.4131, -1.8698, -0.5582, -1.2652, 0.8626, 1.7328, 1.7129, -1.0489, 0.3309, -0.0435, 1.1554], [ 1.2676, -1.9798, -0.7505, -0.2731, -1.1362, 0.5353, -1.3947, 1.6763, -1.4323, 0.5500, 0.3345, -0.1932, 0.2958, -1.0673, 0.1891, 1.8858, 1.9893, -0.4553, 0.4284, -0.4182], [ 0.8169, -1.6031, -1.4634, 0.5036, -0.9926, 1.9319, -1.1357, 1.4299, -0.1737, -0.3567, -0.5193, -0.1565, 1.3256, 0.9212, -1.0795, 0.2933, -0.1425, -0.9356, -0.3506, -1.1891]],

[[-1.4523, 1.6652, 1.8988, -1.5228, -0.4680, 1.7696, -0.0126, 0.0595, -0.9792, 1.3731, -1.0529, 1.3009, -1.9354, -0.4431, -0.3360, -0.8617, 1.6389, 0.5681, 1.4091, -0.4532], [ 0.4200, 1.6801, -0.6350, -0.2937, 0.8192, -1.7547, -0.0023, -0.8332, 1.7256, 1.5671, -1.9655, -0.9337, 1.1646, 0.1157, 0.8171, -0.2420, 1.9833, -1.2515, -1.9892, 1.2424], [ 1.4888, -0.1588, 1.3372, -1.7930, -0.2623, 0.0159, 1.6651, -1.4205, -1.6741, 0.3484, -0.7193, 1.5819, 0.8232, 1.1013, -1.7786, -0.8769, 0.7951, -1.1791, 0.6205, -0.4372], [ 1.8010, -0.6551, -1.6349, -0.1433, -1.6172, 1.7753, -0.3452, -1.8068, 0.2773, 1.0525, 0.2656, -0.0981, -0.6501, -1.4674, 1.1974, 0.1889, -1.1462, 1.2498, 0.0458, -1.5549], [-0.7784, -1.4418, 1.5518, -1.6349, 0.2633, -0.0242, 0.4860, -0.2963, 1.9413, 0.8316, -1.9919, -1.5023, -0.7553, 1.7810, -0.1282, 1.8717, -0.6003, 1.1921, -1.3241, -0.8015]],

[[ 0.0387, -0.0683, -1.3119, -0.6655, 1.3270, 0.4466, -0.5344, 0.1092, -1.6954, -1.7021, 0.2210, 0.0358, 1.7904, -1.5685, -0.0929, -0.3973, 0.2937, 0.8790, 0.6636, 1.0969], [ 1.1867, -1.8680, -0.7894, 0.1756, -1.8204, 0.7961, -1.3391, 0.7083, -1.5092, 1.1195, -1.0322, -0.1478, -0.9416, -1.6585, 1.1229, 0.8414, 1.6447, 0.3158, 1.7880, 0.9119], [ 0.8099, 0.3150, 0.6882, 1.6934, -0.2961, 0.3199, 0.9885, -0.3853, 1.1251, -0.8356, 1.4918, 1.3761, 1.7323, -1.2555, -1.9041, 0.2660, -1.5036, -1.1210, 1.7914, 0.6311], [ 1.9929, 1.1131, 1.9899, 1.0328, -1.4152, 1.5587, -0.4122, -1.5055, 1.6023, -1.2365, -0.9438, 1.2561, -1.5768, 0.6131, 1.9208, 1.6280, -1.4219, 1.7101, 0.0306, 0.4187], [-1.7767, 1.2516, 0.7330, -1.1223, 0.2094, 0.5165, 1.4007, 0.6846, -0.1133, 0.6371, 0.6199, -0.0956, 1.5327, -1.6353, 0.7567, 0.6291, -1.2143, -1.4736, 0.0165, 0.8720]]],

[[[-1.3789, -1.2527, 1.3401, 0.3406, -0.7042, 1.1389, 0.6666, -0.4117, -0.2989, -0.3938, 0.3820, -0.6539, 0.9630, 0.9078, -1.2130, 0.3959, 1.1905, -0.8307, 0.9276, 1.8631], [ 0.6441, -0.0742, -0.2414, 1.8217, 0.9493, 0.8255, -0.7547, -1.2031, -1.3122, 0.5966, 0.7229, -1.6750, 0.0700, -0.6009, 0.2622, 1.1515, -1.0787, -1.4436, -1.4295, 0.3572], [ 0.6079, -0.5907, -0.9739, -0.1204, -1.8401, -0.1114, 0.1417, 1.0373, -0.0542, 1.5353, 0.0925, -1.4203, -0.6187, -0.3825, 1.4880, 0.5443, 1.3337, -1.5342, 0.1816, 0.3702], [-1.8681, 0.5987, -1.0082, -1.9412, -0.0964, -1.7483, 1.1809, 0.5175, 1.9990, 1.1500, -1.7001, 0.5029, -0.1091, -1.1460, 1.7729, -0.5552, -1.4418, 1.4706, -1.9570, 0.4494], [ 1.4411, 0.6376, -0.3329, -0.5400, -0.1568, 0.9053, -0.7919, -0.5293, 0.5903, 0.0579, 1.8277, -1.6401, -0.8383, 0.5593, 0.4174, -0.9361, -0.5755, -0.3629, 1.9672, 1.2225]],

[[-0.5613, -0.3929, 0.2896, -1.7571, 1.9991, -0.4219, -0.7767, -1.8055, -0.3581, -1.7616, -1.6126, 0.2505, -0.2944, 0.4794, -0.3436, 0.2436, 1.0697, -0.7204, 0.8678, 1.1480], [ 1.6177, 0.9862, 0.8313, -1.5676, -0.8029, -0.4710, -0.9152, 1.3049, 1.1056, 1.7808, 0.6238, -1.4056, 1.3417, 0.1611, 0.7123, -1.9622, -1.0206, 1.0915, 1.0982, 1.9980], [-0.3426, 1.1918, 1.3012, -1.3486, -1.1785, 0.6574, 0.4834, -1.5202, -1.4326, -1.0696, 0.7712, 0.6510, 1.4800, 0.4209, -1.5200, -1.4426, -0.4412, -0.0307, 0.9065, -1.8860], [-0.7202, 0.5170, -0.7852, -0.3009, -0.4691, 1.9814, 1.9398, -1.0492, 1.9136, -0.2763, -0.0766, -0.8461, -0.1436, 1.9201, 1.0925, 0.1738, 0.6860, 0.4168, 0.5323, 0.8805], [-0.4933, -1.5355, -1.1386, -1.8716, -0.6593, -0.4551, -1.8895, -0.0941, 0.6320, -1.5665, 0.2567, -1.6547, 0.4524, 0.8572, 0.4672, 0.3772, 0.9701, -0.6117, -0.4405, 0.2073]],

[[ 0.8025, -0.0595, -0.9438, 0.2717, -0.3834, 0.8637, 0.9916, -0.4114, 1.8228, -0.6555, -0.1733, -0.6234, 0.4483, 0.4614, 0.2109, -0.0100, -0.7163, -0.0803, -0.5773, 1.8567], [ 1.1786, 0.0330, 0.9112, 0.1013, -0.1249, -0.5918, -1.8807, -0.6012, -0.4238, 1.9679, -1.3820, 1.6903, -1.8332, -0.8515, 0.0218, 0.7700, -1.7483, 0.7853, -1.9223, 0.1178], [-0.6113, -1.2863, -1.8943, 0.6953, -1.7618, 0.8581, 0.9284, 1.7700, -0.1035, -1.8848, 0.2061, -1.4119, -0.3077, 0.3535, -1.3055, 1.3411, 0.9784, 0.2072, 1.7066, -1.2979], [-1.7509, 1.1533, 1.9451, -0.6432, -1.2304, -1.1662, -1.2911, -0.8734, 0.9899, -1.9781, -0.4201, 1.5957, -1.5980, -0.7309, -1.9116, -0.0167, -1.6116, -1.2348, 0.5920, 1.1657], [-0.1331, 1.6117, 0.5441, -1.8298, -1.6769, 1.5804, -1.6245, 1.8640, -1.0085, 1.5367, 0.4123, 0.5132, -0.4380, 1.5536, -1.8067, 1.9436, -0.7693, -0.0835, 1.0801, 1.2580]]]], device='cuda:0', requires_grad=True), args=(tensor([[[[ 0.8840, 0.7284], [-1.9067, -1.7008], [ 1.8565, -0.3989], [-0.8446, 0.7945], [-1.1669, 0.7130], [-1.8342, -1.8440], [-0.0883, 1.5642], [ 0.7254, 0.5880], [ 0.2307, 1.1799], [ 1.4357, -0.6544]],

[[-1.0316, 1.2807], [ 1.0162, -0.9042], [ 0.5910, 0.3717], [ 0.3884, 0.6074], [-1.0467, -1.9530], [-0.7783, -0.1390], [ 1.5114, -1.0019], [-1.1427, 0.9776], [-0.7653, -0.7106], [ 0.5124, 0.0094]],

[[-0.2061, -1.0028], [ 0.5493, 0.3959], [-0.0231, 0.9015], [-0.1169, 1.0216], [ 1.6647, -0.0388], [-1.0319, 0.4510], [ 1.5703, 0.4492], [-1.9579, -0.3020], [ 0.7635, 1.5684], [ 0.3821, 1.6603]],

[[-0.6688, 0.5106], [-0.4162, 1.6228], [-0.3385, 1.9604], [ 0.7010, -1.9517], [-1.5050, 0.1919], [-1.9553, -1.1527], [-0.1650, 0.0051], [-0.8922, 1.1805], [-0.0818, 1.2613], [ 0.7639, -1.4357]],

[[-1.6299, 0.4255], [-0.6200, -0.6759], [ 1.0267, -1.6362], [ 1.2146, -1.7742], [-1.9981, -1.0791], [-0.7641, 1.6715], [-1.6763, -1.4935], [ 1.0545, 1.9053], [-1.8099, 0.2377], [ 1.8880, -1.7312]],

[[ 1.1423, -1.4683], [ 1.4219, 0.7878], [-0.3497, -0.4911], [ 0.3866, 0.7407], [-1.4957, 0.8446], [ 0.3847, -1.4483], [-0.3230, 0.6764], [-0.1395, -1.7608], [ 0.4753, -0.1717], [-1.9776, -1.1859]],

[[ 0.9461, 1.6932], [ 1.0734, -0.9379], [-0.2483, -1.6817], [ 0.3628, -0.4302], [ 0.8477, -0.1925], [ 1.5683, -1.4538], [-0.2494, -1.6042], [ 0.0142, -1.6684], [-1.6479, -0.3535], [ 1.6242, -1.8027]],

[[-1.2505, -1.3212], [ 1.9157, -0.6191], [-1.7512, 0.5165], [-1.5732, -0.1249], [-0.5898, -0.4158], [-0.4769, -0.3535], [ 0.4879, -0.2919], [-0.1908, 1.6932], [ 1.0552, 1.9789], [ 0.4114, -1.8693]],

[[-0.6473, -0.2875], [ 0.8754, 0.5485], [ 1.7395, 0.5469], [ 1.9870, -0.5567], [-1.5295, 0.1025], [-1.3862, 1.2063], [-0.7479, -0.0594], [ 1.6344, -0.8996], [ 0.5685, -0.6133], [-1.8893, 1.9618]],

[[-1.3969, -1.2584], [-0.5230, 0.7359], [-1.9951, 1.8692], [-1.7383, 0.5308], [ 0.0171, 1.5884], [-0.4782, 1.9707], [ 0.3105, 1.5377], [-0.3061, 1.2858], [-0.0608, 0.7897], [-1.5873, 1.1970]],

[[ 1.3557, -1.8997], [ 0.0401, 1.0429], [ 0.1588, 0.3099], [ 0.8134, -1.7725], [-1.4868, -1.9436], [-1.3589, -1.8519], [ 0.0235, -0.0903], [-0.9190, -0.7169], [-0.4357, -0.9308], [ 1.2138, -0.4648]],

[[ 0.7425, -1.2723], [-1.2882, -0.9645], [ 0.9654, -0.3355], [ 1.1874, 1.9311], [-1.2859, -0.2779], [ 0.0146, 0.6645], [-1.9226, -1.8894], [-1.9399, 0.3529], [-0.9138, 1.9704], [-1.1756, 0.3996]],

[[-0.0967, -1.5186], [-0.5572, 1.4003], [ 1.1635, 0.3844], [-1.8407, 1.2133], [ 0.3440, 0.8430], [ 1.6675, -0.2081], [ 1.5182, -0.0991], [ 0.4312, 1.1988], [ 1.0257, 0.9181], [-1.3318, 0.0113]]],

[[[ 0.5229, 1.7186], [ 0.2447, -1.7722], [ 1.9558, 0.7897], [-0.2866, 1.7106], [-1.8656, -0.6073], [ 0.2541, -0.5758], [ 0.7862, -0.7967], [ 0.2434, -0.2500], [ 1.9724, -1.6783], [-1.2378, -1.8009]],

[[-0.2075, 1.0700], [-1.8012, -0.7954], [ 0.7769, -0.3403], [ 1.7522, -1.6351], [ 1.1443, -0.0109], [ 1.1499, 1.1863], [ 0.5536, -1.2255], [ 1.1178, -1.0296], [ 0.5785, -1.0306], [ 0.4561, 0.3083]],

[[ 1.8243, -0.4109], [-0.8581, -0.9415], [ 1.6291, 1.4080], [ 0.2663, -1.4106], [-1.3930, 0.4565], [ 1.3747, -1.4464], [-0.5173, 0.6275], [-1.4881, -0.0791], [-1.5300, -0.0376], [-0.7357, 1.3118]],

[[ 0.4213, -0.4424], [ 0.4718, 0.8544], [ 0.6342, 0.5930], [-1.4592, 0.9840], [-0.2018, -1.5982], [ 0.2531, 0.1849], [ 0.7382, -1.9418], [-0.8996, 0.3930], [ 0.3761, 0.0904], [-1.3371, 0.9920]],

[[ 0.8061, -0.4647], [-1.5188, 1.0175], [ 1.3355, -0.9962], [-1.2889, -1.5765], [-0.2926, 1.8030], [-1.2162, 1.3498], [-1.0174, -0.0172], [ 1.9736, -0.9195], [-0.5556, -0.3043], [-0.8305, 0.3111]],

[[-0.1680, 1.6404], [ 0.2090, 0.4003], [ 1.7559, -0.1290], [-1.2025, 1.6680], [-1.0993, 1.4024], [-1.6462, -1.7576], [ 0.3758, -0.5544], [ 1.7620, 0.0187], [ 0.7033, 1.0926], [ 1.8603, 1.8435]],

[[-0.2956, -0.1107], [-0.9312, 0.3183], [-0.9891, -1.5630], [-1.4147, -0.0128], [-1.5586, 1.3915], [ 1.1812, -1.4658], [-1.8646, 0.9345], [-0.0903, -0.1244], [ 0.4643, -0.7193], [ 0.2759, -0.7039]],

[[ 1.6475, -0.4765], [ 1.7192, 0.9523], [-0.3220, -0.6531], [ 0.3992, 1.8709], [-1.9119, 1.7273], [ 0.7452, 1.1067], [-0.1234, 0.8440], [ 1.7273, -0.4334], [ 0.4186, -0.4803], [-1.5332, -0.1467]],

[[ 1.7229, 0.9690], [-1.8442, -1.0074], [ 0.4095, 0.1997], [ 1.2701, -1.8232], [-1.7041, -0.3767], [-0.7734, -0.2795], [-1.3943, -0.6950], [ 0.2118, 1.2398], [-1.0847, -0.2509], [-0.7644, 1.8773]],

[[ 0.3199, -1.6492], [ 0.4723, -0.6576], [ 0.3100, -1.8890], [ 1.3837, -0.1127], [ 0.4850, -1.1013], [-1.5918, -0.1211], [ 0.1576, -0.7470], [-0.6888, -1.3469], [-0.9145, 0.3586], [-1.7643, 0.4087]],

[[-0.7426, 0.7931], [-1.1826, -1.7763], [ 0.4773, -1.5527], [ 1.6368, -0.7878], [ 1.8296, -0.6230], [ 1.6000, -0.3928], [-0.2718, -1.6651], [ 1.1806, 0.5856], [ 1.9963, -0.4580], [-0.0518, -0.0662]],

[[-1.5275, -0.2008], [-1.0342, -1.4198], [ 1.5619, -0.8835], [ 0.4056, 0.8474], [ 0.6425, 1.1953], [ 1.0211, -1.4575], [ 1.4458, 0.1762], [-0.9242, -1.8079], [-1.2674, -0.0075], [-1.2202, -1.8033]],

[[ 0.8573, 0.8268], [ 1.0164, 1.8607], [-0.3526, -0.6622], [-0.1924, 0.5389], [ 1.6591, 1.8509], [ 1.1368, 0.9931], [-0.0709, 1.3218], [ 1.5008, 0.9046], [-1.4040, 1.6321], [ 1.0302, -0.2132]]]], device='cuda:0', requires_grad=True), 2, 1, False), kwargs={}, broadcasts_input=False, name=''):

Caused by sample input at index 14: SampleInput(input=Tensor[size=(2, 3, 5, 20), device="cuda:0", dtype=torch.float32], args=(Tensor[size=(2, 13, 10, 2), device="cuda:0", dtype=torch.float32],2,1,False), kwargs={}, broadcasts_input=False, name='')

To execute this test, run the following from the base repo dir: PYTORCH_OPINFO_SAMPLE_INPUT_INDEX=14 python test/test_ops_jit.py TestJitCUDA.test_variant_consistency_jit_grid_sampler_2d_cuda_float32

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0

Root Cause

This test was disabled because it is failing in CI. See recent examples and the most recent trunk workflow logs.

Code Example

Traceback (most recent call last):
  File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 109, in test_variant_consistency_jit
    self.indiv_variant_test_jit(
  File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 157, in indiv_variant_test_jit
    check_against_reference(
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_jit.py", line 140, in check_against_reference
    self.assertEqual(g2, g2_test, atol=5e-4, rtol=1e-4)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 4440, in assertEqual
    raise error_metas.pop()[0].to_error(  # type: ignore[index]
AssertionError: Tensor-likes are not close!

Mismatched elements: 84 / 520 (16.2%)
Greatest absolute difference: 0.005322886630892754 at index (0, 4, 8, 0) (up to 0.0005 allowed)
Greatest relative difference: 11.676852226257324 at index (0, 8, 9, 0) (up to 0.0001 allowed)

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 1175, in test_wrapper
    return test(*args, **kwargs)
  File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 120, in test_variant_consistency_jit
    raise Exception(variant_error_info) from e  # noqa: TRY002
Exception: 
                 Error testing grid_sampler_2d function variant
                 with dtype: torch.float32
                 with inputs SampleInput(input=tensor([[[[-0.7560, -0.1443, -1.3317,  1.0707, -0.0536,  1.2746, -1.0104,
     1.7234,  0.4455,  0.2850,  0.4191,  1.4160, -1.6165,  1.2780,
    -1.4350,  0.3450, -1.6000,  0.3590,  1.1890,  1.1320],
   [-1.2507, -0.5309, -1.4537,  0.7853,  1.7886, -0.0907,  1.6612,
     1.2864,  0.1156, -0.8906, -0.7386, -1.8012, -1.8938,  0.8986,
     0.0282, -0.9127,  0.3131, -1.2557,  0.7019,  1.8979],
   [-0.9708,  1.1602, -1.4717, -1.8265, -1.3409,  1.7040,  0.1004,
    -0.4273,  0.3805,  0.4131, -1.8698, -0.5582, -1.2652,  0.8626,
     1.7328,  1.7129, -1.0489,  0.3309, -0.0435,  1.1554],
   [ 1.2676, -1.9798, -0.7505, -0.2731, -1.1362,  0.5353, -1.3947,
     1.6763, -1.4323,  0.5500,  0.3345, -0.1932,  0.2958, -1.0673,
     0.1891,  1.8858,  1.9893, -0.4553,  0.4284, -0.4182],
   [ 0.8169, -1.6031, -1.4634,  0.5036, -0.9926,  1.9319, -1.1357,
     1.4299, -0.1737, -0.3567, -0.5193, -0.1565,  1.3256,  0.9212,
    -1.0795,  0.2933, -0.1425, -0.9356, -0.3506, -1.1891]],

  [[-1.4523,  1.6652,  1.8988, -1.5228, -0.4680,  1.7696, -0.0126,
     0.0595, -0.9792,  1.3731, -1.0529,  1.3009, -1.9354, -0.4431,
    -0.3360, -0.8617,  1.6389,  0.5681,  1.4091, -0.4532],
   [ 0.4200,  1.6801, -0.6350, -0.2937,  0.8192, -1.7547, -0.0023,
    -0.8332,  1.7256,  1.5671, -1.9655, -0.9337,  1.1646,  0.1157,
     0.8171, -0.2420,  1.9833, -1.2515, -1.9892,  1.2424],
   [ 1.4888, -0.1588,  1.3372, -1.7930, -0.2623,  0.0159,  1.6651,
    -1.4205, -1.6741,  0.3484, -0.7193,  1.5819,  0.8232,  1.1013,
    -1.7786, -0.8769,  0.7951, -1.1791,  0.6205, -0.4372],
   [ 1.8010, -0.6551, -1.6349, -0.1433, -1.6172,  1.7753, -0.3452,
    -1.8068,  0.2773,  1.0525,  0.2656, -0.0981, -0.6501, -1.4674,
     1.1974,  0.1889, -1.1462,  1.2498,  0.0458, -1.5549],
   [-0.7784, -1.4418,  1.5518, -1.6349,  0.2633, -0.0242,  0.4860,
    -0.2963,  1.9413,  0.8316, -1.9919, -1.5023, -0.7553,  1.7810,
    -0.1282,  1.8717, -0.6003,  1.1921, -1.3241, -0.8015]],

  [[ 0.0387, -0.0683, -1.3119, -0.6655,  1.3270,  0.4466, -0.5344,
     0.1092, -1.6954, -1.7021,  0.2210,  0.0358,  1.7904, -1.5685,
    -0.0929, -0.3973,  0.2937,  0.8790,  0.6636,  1.0969],
   [ 1.1867, -1.8680, -0.7894,  0.1756, -1.8204,  0.7961, -1.3391,
     0.7083, -1.5092,  1.1195, -1.0322, -0.1478, -0.9416, -1.6585,
     1.1229,  0.8414,  1.6447,  0.3158,  1.7880,  0.9119],
   [ 0.8099,  0.3150,  0.6882,  1.6934, -0.2961,  0.3199,  0.9885,
    -0.3853,  1.1251, -0.8356,  1.4918,  1.3761,  1.7323, -1.2555,
    -1.9041,  0.2660, -1.5036, -1.1210,  1.7914,  0.6311],
   [ 1.9929,  1.1131,  1.9899,  1.0328, -1.4152,  1.5587, -0.4122,
    -1.5055,  1.6023, -1.2365, -0.9438,  1.2561, -1.5768,  0.6131,
     1.9208,  1.6280, -1.4219,  1.7101,  0.0306,  0.4187],
   [-1.7767,  1.2516,  0.7330, -1.1223,  0.2094,  0.5165,  1.4007,
     0.6846, -0.1133,  0.6371,  0.6199, -0.0956,  1.5327, -1.6353,
     0.7567,  0.6291, -1.2143, -1.4736,  0.0165,  0.8720]]],


 [[[-1.3789, -1.2527,  1.3401,  0.3406, -0.7042,  1.1389,  0.6666,
    -0.4117, -0.2989, -0.3938,  0.3820, -0.6539,  0.9630,  0.9078,
    -1.2130,  0.3959,  1.1905, -0.8307,  0.9276,  1.8631],
   [ 0.6441, -0.0742, -0.2414,  1.8217,  0.9493,  0.8255, -0.7547,
    -1.2031, -1.3122,  0.5966,  0.7229, -1.6750,  0.0700, -0.6009,
     0.2622,  1.1515, -1.0787, -1.4436, -1.4295,  0.3572],
   [ 0.6079, -0.5907, -0.9739, -0.1204, -1.8401, -0.1114,  0.1417,
     1.0373, -0.0542,  1.5353,  0.0925, -1.4203, -0.6187, -0.3825,
     1.4880,  0.5443,  1.3337, -1.5342,  0.1816,  0.3702],
   [-1.8681,  0.5987, -1.0082, -1.9412, -0.0964, -1.7483,  1.1809,
     0.5175,  1.9990,  1.1500, -1.7001,  0.5029, -0.1091, -1.1460,
     1.7729, -0.5552, -1.4418,  1.4706, -1.9570,  0.4494],
   [ 1.4411,  0.6376, -0.3329, -0.5400, -0.1568,  0.9053, -0.7919,
    -0.5293,  0.5903,  0.0579,  1.8277, -1.6401, -0.8383,  0.5593,
     0.4174, -0.9361, -0.5755, -0.3629,  1.9672,  1.2225]],

  [[-0.5613, -0.3929,  0.2896, -1.7571,  1.9991, -0.4219, -0.7767,
    -1.8055, -0.3581, -1.7616, -1.6126,  0.2505, -0.2944,  0.4794,
    -0.3436,  0.2436,  1.0697, -0.7204,  0.8678,  1.1480],
   [ 1.6177,  0.9862,  0.8313, -1.5676, -0.8029, -0.4710, -0.9152,
     1.3049,  1.1056,  1.7808,  0.6238, -1.4056,  1.3417,  0.1611,
     0.7123, -1.9622, -1.0206,  1.0915,  1.0982,  1.9980],
   [-0.3426,  1.1918,  1.3012, -1.3486, -1.1785,  0.6574,  0.4834,
    -1.5202, -1.4326, -1.0696,  0.7712,  0.6510,  1.4800,  0.4209,
    -1.5200, -1.4426, -0.4412, -0.0307,  0.9065, -1.8860],
   [-0.7202,  0.5170, -0.7852, -0.3009, -0.4691,  1.9814,  1.9398,
    -1.0492,  1.9136, -0.2763, -0.0766, -0.8461, -0.1436,  1.9201,
     1.0925,  0.1738,  0.6860,  0.4168,  0.5323,  0.8805],
   [-0.4933, -1.5355, -1.1386, -1.8716, -0.6593, -0.4551, -1.8895,
    -0.0941,  0.6320, -1.5665,  0.2567, -1.6547,  0.4524,  0.8572,
     0.4672,  0.3772,  0.9701, -0.6117, -0.4405,  0.2073]],

  [[ 0.8025, -0.0595, -0.9438,  0.2717, -0.3834,  0.8637,  0.9916,
    -0.4114,  1.8228, -0.6555, -0.1733, -0.6234,  0.4483,  0.4614,
     0.2109, -0.0100, -0.7163, -0.0803, -0.5773,  1.8567],
   [ 1.1786,  0.0330,  0.9112,  0.1013, -0.1249, -0.5918, -1.8807,
    -0.6012, -0.4238,  1.9679, -1.3820,  1.6903, -1.8332, -0.8515,
     0.0218,  0.7700, -1.7483,  0.7853, -1.9223,  0.1178],
   [-0.6113, -1.2863, -1.8943,  0.6953, -1.7618,  0.8581,  0.9284,
     1.7700, -0.1035, -1.8848,  0.2061, -1.4119, -0.3077,  0.3535,
    -1.3055,  1.3411,  0.9784,  0.2072,  1.7066, -1.2979],
   [-1.7509,  1.1533,  1.9451, -0.6432, -1.2304, -1.1662, -1.2911,
    -0.8734,  0.9899, -1.9781, -0.4201,  1.5957, -1.5980, -0.7309,
    -1.9116, -0.0167, -1.6116, -1.2348,  0.5920,  1.1657],
   [-0.1331,  1.6117,  0.5441, -1.8298, -1.6769,  1.5804, -1.6245,
     1.8640, -1.0085,  1.5367,  0.4123,  0.5132, -0.4380,  1.5536,
    -1.8067,  1.9436, -0.7693, -0.0835,  1.0801,  1.2580]]]],
device='cuda:0', requires_grad=True), args=(tensor([[[[ 0.8840,  0.7284],
   [-1.9067, -1.7008],
   [ 1.8565, -0.3989],
   [-0.8446,  0.7945],
   [-1.1669,  0.7130],
   [-1.8342, -1.8440],
   [-0.0883,  1.5642],
   [ 0.7254,  0.5880],
   [ 0.2307,  1.1799],
   [ 1.4357, -0.6544]],

  [[-1.0316,  1.2807],
   [ 1.0162, -0.9042],
   [ 0.5910,  0.3717],
   [ 0.3884,  0.6074],
   [-1.0467, -1.9530],
   [-0.7783, -0.1390],
   [ 1.5114, -1.0019],
   [-1.1427,  0.9776],
   [-0.7653, -0.7106],
   [ 0.5124,  0.0094]],

  [[-0.2061, -1.0028],
   [ 0.5493,  0.3959],
   [-0.0231,  0.9015],
   [-0.1169,  1.0216],
   [ 1.6647, -0.0388],
   [-1.0319,  0.4510],
   [ 1.5703,  0.4492],
   [-1.9579, -0.3020],
   [ 0.7635,  1.5684],
   [ 0.3821,  1.6603]],

  [[-0.6688,  0.5106],
   [-0.4162,  1.6228],
   [-0.3385,  1.9604],
   [ 0.7010, -1.9517],
   [-1.5050,  0.1919],
   [-1.9553, -1.1527],
   [-0.1650,  0.0051],
   [-0.8922,  1.1805],
   [-0.0818,  1.2613],
   [ 0.7639, -1.4357]],

  [[-1.6299,  0.4255],
   [-0.6200, -0.6759],
   [ 1.0267, -1.6362],
   [ 1.2146, -1.7742],
   [-1.9981, -1.0791],
   [-0.7641,  1.6715],
   [-1.6763, -1.4935],
   [ 1.0545,  1.9053],
   [-1.8099,  0.2377],
   [ 1.8880, -1.7312]],

  [[ 1.1423, -1.4683],
   [ 1.4219,  0.7878],
   [-0.3497, -0.4911],
   [ 0.3866,  0.7407],
   [-1.4957,  0.8446],
   [ 0.3847, -1.4483],
   [-0.3230,  0.6764],
   [-0.1395, -1.7608],
   [ 0.4753, -0.1717],
   [-1.9776, -1.1859]],

  [[ 0.9461,  1.6932],
   [ 1.0734, -0.9379],
   [-0.2483, -1.6817],
   [ 0.3628, -0.4302],
   [ 0.8477, -0.1925],
   [ 1.5683, -1.4538],
   [-0.2494, -1.6042],
   [ 0.0142, -1.6684],
   [-1.6479, -0.3535],
   [ 1.6242, -1.8027]],

  [[-1.2505, -1.3212],
   [ 1.9157, -0.6191],
   [-1.7512,  0.5165],
   [-1.5732, -0.1249],
   [-0.5898, -0.4158],
   [-0.4769, -0.3535],
   [ 0.4879, -0.2919],
   [-0.1908,  1.6932],
   [ 1.0552,  1.9789],
   [ 0.4114, -1.8693]],

  [[-0.6473, -0.2875],
   [ 0.8754,  0.5485],
   [ 1.7395,  0.5469],
   [ 1.9870, -0.5567],
   [-1.5295,  0.1025],
   [-1.3862,  1.2063],
   [-0.7479, -0.0594],
   [ 1.6344, -0.8996],
   [ 0.5685, -0.6133],
   [-1.8893,  1.9618]],

  [[-1.3969, -1.2584],
   [-0.5230,  0.7359],
   [-1.9951,  1.8692],
   [-1.7383,  0.5308],
   [ 0.0171,  1.5884],
   [-0.4782,  1.9707],
   [ 0.3105,  1.5377],
   [-0.3061,  1.2858],
   [-0.0608,  0.7897],
   [-1.5873,  1.1970]],

  [[ 1.3557, -1.8997],
   [ 0.0401,  1.0429],
   [ 0.1588,  0.3099],
   [ 0.8134, -1.7725],
   [-1.4868, -1.9436],
   [-1.3589, -1.8519],
   [ 0.0235, -0.0903],
   [-0.9190, -0.7169],
   [-0.4357, -0.9308],
   [ 1.2138, -0.4648]],

  [[ 0.7425, -1.2723],
   [-1.2882, -0.9645],
   [ 0.9654, -0.3355],
   [ 1.1874,  1.9311],
   [-1.2859, -0.2779],
   [ 0.0146,  0.6645],
   [-1.9226, -1.8894],
   [-1.9399,  0.3529],
   [-0.9138,  1.9704],
   [-1.1756,  0.3996]],

  [[-0.0967, -1.5186],
   [-0.5572,  1.4003],
   [ 1.1635,  0.3844],
   [-1.8407,  1.2133],
   [ 0.3440,  0.8430],
   [ 1.6675, -0.2081],
   [ 1.5182, -0.0991],
   [ 0.4312,  1.1988],
   [ 1.0257,  0.9181],
   [-1.3318,  0.0113]]],


 [[[ 0.5229,  1.7186],
   [ 0.2447, -1.7722],
   [ 1.9558,  0.7897],
   [-0.2866,  1.7106],
   [-1.8656, -0.6073],
   [ 0.2541, -0.5758],
   [ 0.7862, -0.7967],
   [ 0.2434, -0.2500],
   [ 1.9724, -1.6783],
   [-1.2378, -1.8009]],

  [[-0.2075,  1.0700],
   [-1.8012, -0.7954],
   [ 0.7769, -0.3403],
   [ 1.7522, -1.6351],
   [ 1.1443, -0.0109],
   [ 1.1499,  1.1863],
   [ 0.5536, -1.2255],
   [ 1.1178, -1.0296],
   [ 0.5785, -1.0306],
   [ 0.4561,  0.3083]],

  [[ 1.8243, -0.4109],
   [-0.8581, -0.9415],
   [ 1.6291,  1.4080],
   [ 0.2663, -1.4106],
   [-1.3930,  0.4565],
   [ 1.3747, -1.4464],
   [-0.5173,  0.6275],
   [-1.4881, -0.0791],
   [-1.5300, -0.0376],
   [-0.7357,  1.3118]],

  [[ 0.4213, -0.4424],
   [ 0.4718,  0.8544],
   [ 0.6342,  0.5930],
   [-1.4592,  0.9840],
   [-0.2018, -1.5982],
   [ 0.2531,  0.1849],
   [ 0.7382, -1.9418],
   [-0.8996,  0.3930],
   [ 0.3761,  0.0904],
   [-1.3371,  0.9920]],

  [[ 0.8061, -0.4647],
   [-1.5188,  1.0175],
   [ 1.3355, -0.9962],
   [-1.2889, -1.5765],
   [-0.2926,  1.8030],
   [-1.2162,  1.3498],
   [-1.0174, -0.0172],
   [ 1.9736, -0.9195],
   [-0.5556, -0.3043],
   [-0.8305,  0.3111]],

  [[-0.1680,  1.6404],
   [ 0.2090,  0.4003],
   [ 1.7559, -0.1290],
   [-1.2025,  1.6680],
   [-1.0993,  1.4024],
   [-1.6462, -1.7576],
   [ 0.3758, -0.5544],
   [ 1.7620,  0.0187],
   [ 0.7033,  1.0926],
   [ 1.8603,  1.8435]],

  [[-0.2956, -0.1107],
   [-0.9312,  0.3183],
   [-0.9891, -1.5630],
   [-1.4147, -0.0128],
   [-1.5586,  1.3915],
   [ 1.1812, -1.4658],
   [-1.8646,  0.9345],
   [-0.0903, -0.1244],
   [ 0.4643, -0.7193],
   [ 0.2759, -0.7039]],

  [[ 1.6475, -0.4765],
   [ 1.7192,  0.9523],
   [-0.3220, -0.6531],
   [ 0.3992,  1.8709],
   [-1.9119,  1.7273],
   [ 0.7452,  1.1067],
   [-0.1234,  0.8440],
   [ 1.7273, -0.4334],
   [ 0.4186, -0.4803],
   [-1.5332, -0.1467]],

  [[ 1.7229,  0.9690],
   [-1.8442, -1.0074],
   [ 0.4095,  0.1997],
   [ 1.2701, -1.8232],
   [-1.7041, -0.3767],
   [-0.7734, -0.2795],
   [-1.3943, -0.6950],
   [ 0.2118,  1.2398],
   [-1.0847, -0.2509],
   [-0.7644,  1.8773]],

  [[ 0.3199, -1.6492],
   [ 0.4723, -0.6576],
   [ 0.3100, -1.8890],
   [ 1.3837, -0.1127],
   [ 0.4850, -1.1013],
   [-1.5918, -0.1211],
   [ 0.1576, -0.7470],
   [-0.6888, -1.3469],
   [-0.9145,  0.3586],
   [-1.7643,  0.4087]],

  [[-0.7426,  0.7931],
   [-1.1826, -1.7763],
   [ 0.4773, -1.5527],
   [ 1.6368, -0.7878],
   [ 1.8296, -0.6230],
   [ 1.6000, -0.3928],
   [-0.2718, -1.6651],
   [ 1.1806,  0.5856],
   [ 1.9963, -0.4580],
   [-0.0518, -0.0662]],

  [[-1.5275, -0.2008],
   [-1.0342, -1.4198],
   [ 1.5619, -0.8835],
   [ 0.4056,  0.8474],
   [ 0.6425,  1.1953],
   [ 1.0211, -1.4575],
   [ 1.4458,  0.1762],
   [-0.9242, -1.8079],
   [-1.2674, -0.0075],
   [-1.2202, -1.8033]],

  [[ 0.8573,  0.8268],
   [ 1.0164,  1.8607],
   [-0.3526, -0.6622],
   [-0.1924,  0.5389],
   [ 1.6591,  1.8509],
   [ 1.1368,  0.9931],
   [-0.0709,  1.3218],
   [ 1.5008,  0.9046],
   [-1.4040,  1.6321],
   [ 1.0302, -0.2132]]]], device='cuda:0', requires_grad=True), 2, 1, False), kwargs={}, broadcasts_input=False, name=''):


The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 3443, in wrapper
    method(*args, **kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 3443, in wrapper
    method(*args, **kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 439, in instantiated_test
    result = test(self, **param_kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 1820, in wrapper
    fn(*args, **kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 1187, in test_wrapper
    raise e_tracked from e
Exception: 
                 Error testing grid_sampler_2d function variant
                 with dtype: torch.float32
                 with inputs SampleInput(input=tensor([[[[-0.7560, -0.1443, -1.3317,  1.0707, -0.0536,  1.2746, -1.0104,
     1.7234,  0.4455,  0.2850,  0.4191,  1.4160, -1.6165,  1.2780,
    -1.4350,  0.3450, -1.6000,  0.3590,  1.1890,  1.1320],
   [-1.2507, -0.5309, -1.4537,  0.7853,  1.7886, -0.0907,  1.6612,
     1.2864,  0.1156, -0.8906, -0.7386, -1.8012, -1.8938,  0.8986,
     0.0282, -0.9127,  0.3131, -1.2557,  0.7019,  1.8979],
   [-0.9708,  1.1602, -1.4717, -1.8265, -1.3409,  1.7040,  0.1004,
    -0.4273,  0.3805,  0.4131, -1.8698, -0.5582, -1.2652,  0.8626,
     1.7328,  1.7129, -1.0489,  0.3309, -0.0435,  1.1554],
   [ 1.2676, -1.9798, -0.7505, -0.2731, -1.1362,  0.5353, -1.3947,
     1.6763, -1.4323,  0.5500,  0.3345, -0.1932,  0.2958, -1.0673,
     0.1891,  1.8858,  1.9893, -0.4553,  0.4284, -0.4182],
   [ 0.8169, -1.6031, -1.4634,  0.5036, -0.9926,  1.9319, -1.1357,
     1.4299, -0.1737, -0.3567, -0.5193, -0.1565,  1.3256,  0.9212,
    -1.0795,  0.2933, -0.1425, -0.9356, -0.3506, -1.1891]],

  [[-1.4523,  1.6652,  1.8988, -1.5228, -0.4680,  1.7696, -0.0126,
     0.0595, -0.9792,  1.3731, -1.0529,  1.3009, -1.9354, -0.4431,
    -0.3360, -0.8617,  1.6389,  0.5681,  1.4091, -0.4532],
   [ 0.4200,  1.6801, -0.6350, -0.2937,  0.8192, -1.7547, -0.0023,
    -0.8332,  1.7256,  1.5671, -1.9655, -0.9337,  1.1646,  0.1157,
     0.8171, -0.2420,  1.9833, -1.2515, -1.9892,  1.2424],
   [ 1.4888, -0.1588,  1.3372, -1.7930, -0.2623,  0.0159,  1.6651,
    -1.4205, -1.6741,  0.3484, -0.7193,  1.5819,  0.8232,  1.1013,
    -1.7786, -0.8769,  0.7951, -1.1791,  0.6205, -0.4372],
   [ 1.8010, -0.6551, -1.6349, -0.1433, -1.6172,  1.7753, -0.3452,
    -1.8068,  0.2773,  1.0525,  0.2656, -0.0981, -0.6501, -1.4674,
     1.1974,  0.1889, -1.1462,  1.2498,  0.0458, -1.5549],
   [-0.7784, -1.4418,  1.5518, -1.6349,  0.2633, -0.0242,  0.4860,
    -0.2963,  1.9413,  0.8316, -1.9919, -1.5023, -0.7553,  1.7810,
    -0.1282,  1.8717, -0.6003,  1.1921, -1.3241, -0.8015]],

  [[ 0.0387, -0.0683, -1.3119, -0.6655,  1.3270,  0.4466, -0.5344,
     0.1092, -1.6954, -1.7021,  0.2210,  0.0358,  1.7904, -1.5685,
    -0.0929, -0.3973,  0.2937,  0.8790,  0.6636,  1.0969],
   [ 1.1867, -1.8680, -0.7894,  0.1756, -1.8204,  0.7961, -1.3391,
     0.7083, -1.5092,  1.1195, -1.0322, -0.1478, -0.9416, -1.6585,
     1.1229,  0.8414,  1.6447,  0.3158,  1.7880,  0.9119],
   [ 0.8099,  0.3150,  0.6882,  1.6934, -0.2961,  0.3199,  0.9885,
    -0.3853,  1.1251, -0.8356,  1.4918,  1.3761,  1.7323, -1.2555,
    -1.9041,  0.2660, -1.5036, -1.1210,  1.7914,  0.6311],
   [ 1.9929,  1.1131,  1.9899,  1.0328, -1.4152,  1.5587, -0.4122,
    -1.5055,  1.6023, -1.2365, -0.9438,  1.2561, -1.5768,  0.6131,
     1.9208,  1.6280, -1.4219,  1.7101,  0.0306,  0.4187],
   [-1.7767,  1.2516,  0.7330, -1.1223,  0.2094,  0.5165,  1.4007,
     0.6846, -0.1133,  0.6371,  0.6199, -0.0956,  1.5327, -1.6353,
     0.7567,  0.6291, -1.2143, -1.4736,  0.0165,  0.8720]]],


 [[[-1.3789, -1.2527,  1.3401,  0.3406, -0.7042,  1.1389,  0.6666,
    -0.4117, -0.2989, -0.3938,  0.3820, -0.6539,  0.9630,  0.9078,
    -1.2130,  0.3959,  1.1905, -0.8307,  0.9276,  1.8631],
   [ 0.6441, -0.0742, -0.2414,  1.8217,  0.9493,  0.8255, -0.7547,
    -1.2031, -1.3122,  0.5966,  0.7229, -1.6750,  0.0700, -0.6009,
     0.2622,  1.1515, -1.0787, -1.4436, -1.4295,  0.3572],
   [ 0.6079, -0.5907, -0.9739, -0.1204, -1.8401, -0.1114,  0.1417,
     1.0373, -0.0542,  1.5353,  0.0925, -1.4203, -0.6187, -0.3825,
     1.4880,  0.5443,  1.3337, -1.5342,  0.1816,  0.3702],
   [-1.8681,  0.5987, -1.0082, -1.9412, -0.0964, -1.7483,  1.1809,
     0.5175,  1.9990,  1.1500, -1.7001,  0.5029, -0.1091, -1.1460,
     1.7729, -0.5552, -1.4418,  1.4706, -1.9570,  0.4494],
   [ 1.4411,  0.6376, -0.3329, -0.5400, -0.1568,  0.9053, -0.7919,
    -0.5293,  0.5903,  0.0579,  1.8277, -1.6401, -0.8383,  0.5593,
     0.4174, -0.9361, -0.5755, -0.3629,  1.9672,  1.2225]],

  [[-0.5613, -0.3929,  0.2896, -1.7571,  1.9991, -0.4219, -0.7767,
    -1.8055, -0.3581, -1.7616, -1.6126,  0.2505, -0.2944,  0.4794,
    -0.3436,  0.2436,  1.0697, -0.7204,  0.8678,  1.1480],
   [ 1.6177,  0.9862,  0.8313, -1.5676, -0.8029, -0.4710, -0.9152,
     1.3049,  1.1056,  1.7808,  0.6238, -1.4056,  1.3417,  0.1611,
     0.7123, -1.9622, -1.0206,  1.0915,  1.0982,  1.9980],
   [-0.3426,  1.1918,  1.3012, -1.3486, -1.1785,  0.6574,  0.4834,
    -1.5202, -1.4326, -1.0696,  0.7712,  0.6510,  1.4800,  0.4209,
    -1.5200, -1.4426, -0.4412, -0.0307,  0.9065, -1.8860],
   [-0.7202,  0.5170, -0.7852, -0.3009, -0.4691,  1.9814,  1.9398,
    -1.0492,  1.9136, -0.2763, -0.0766, -0.8461, -0.1436,  1.9201,
     1.0925,  0.1738,  0.6860,  0.4168,  0.5323,  0.8805],
   [-0.4933, -1.5355, -1.1386, -1.8716, -0.6593, -0.4551, -1.8895,
    -0.0941,  0.6320, -1.5665,  0.2567, -1.6547,  0.4524,  0.8572,
     0.4672,  0.3772,  0.9701, -0.6117, -0.4405,  0.2073]],

  [[ 0.8025, -0.0595, -0.9438,  0.2717, -0.3834,  0.8637,  0.9916,
    -0.4114,  1.8228, -0.6555, -0.1733, -0.6234,  0.4483,  0.4614,
     0.2109, -0.0100, -0.7163, -0.0803, -0.5773,  1.8567],
   [ 1.1786,  0.0330,  0.9112,  0.1013, -0.1249, -0.5918, -1.8807,
    -0.6012, -0.4238,  1.9679, -1.3820,  1.6903, -1.8332, -0.8515,
     0.0218,  0.7700, -1.7483,  0.7853, -1.9223,  0.1178],
   [-0.6113, -1.2863, -1.8943,  0.6953, -1.7618,  0.8581,  0.9284,
     1.7700, -0.1035, -1.8848,  0.2061, -1.4119, -0.3077,  0.3535,
    -1.3055,  1.3411,  0.9784,  0.2072,  1.7066, -1.2979],
   [-1.7509,  1.1533,  1.9451, -0.6432, -1.2304, -1.1662, -1.2911,
    -0.8734,  0.9899, -1.9781, -0.4201,  1.5957, -1.5980, -0.7309,
    -1.9116, -0.0167, -1.6116, -1.2348,  0.5920,  1.1657],
   [-0.1331,  1.6117,  0.5441, -1.8298, -1.6769,  1.5804, -1.6245,
     1.8640, -1.0085,  1.5367,  0.4123,  0.5132, -0.4380,  1.5536,
    -1.8067,  1.9436, -0.7693, -0.0835,  1.0801,  1.2580]]]],
device='cuda:0', requires_grad=True), args=(tensor([[[[ 0.8840,  0.7284],
   [-1.9067, -1.7008],
   [ 1.8565, -0.3989],
   [-0.8446,  0.7945],
   [-1.1669,  0.7130],
   [-1.8342, -1.8440],
   [-0.0883,  1.5642],
   [ 0.7254,  0.5880],
   [ 0.2307,  1.1799],
   [ 1.4357, -0.6544]],

  [[-1.0316,  1.2807],
   [ 1.0162, -0.9042],
   [ 0.5910,  0.3717],
   [ 0.3884,  0.6074],
   [-1.0467, -1.9530],
   [-0.7783, -0.1390],
   [ 1.5114, -1.0019],
   [-1.1427,  0.9776],
   [-0.7653, -0.7106],
   [ 0.5124,  0.0094]],

  [[-0.2061, -1.0028],
   [ 0.5493,  0.3959],
   [-0.0231,  0.9015],
   [-0.1169,  1.0216],
   [ 1.6647, -0.0388],
   [-1.0319,  0.4510],
   [ 1.5703,  0.4492],
   [-1.9579, -0.3020],
   [ 0.7635,  1.5684],
   [ 0.3821,  1.6603]],

  [[-0.6688,  0.5106],
   [-0.4162,  1.6228],
   [-0.3385,  1.9604],
   [ 0.7010, -1.9517],
   [-1.5050,  0.1919],
   [-1.9553, -1.1527],
   [-0.1650,  0.0051],
   [-0.8922,  1.1805],
   [-0.0818,  1.2613],
   [ 0.7639, -1.4357]],

  [[-1.6299,  0.4255],
   [-0.6200, -0.6759],
   [ 1.0267, -1.6362],
   [ 1.2146, -1.7742],
   [-1.9981, -1.0791],
   [-0.7641,  1.6715],
   [-1.6763, -1.4935],
   [ 1.0545,  1.9053],
   [-1.8099,  0.2377],
   [ 1.8880, -1.7312]],

  [[ 1.1423, -1.4683],
   [ 1.4219,  0.7878],
   [-0.3497, -0.4911],
   [ 0.3866,  0.7407],
   [-1.4957,  0.8446],
   [ 0.3847, -1.4483],
   [-0.3230,  0.6764],
   [-0.1395, -1.7608],
   [ 0.4753, -0.1717],
   [-1.9776, -1.1859]],

  [[ 0.9461,  1.6932],
   [ 1.0734, -0.9379],
   [-0.2483, -1.6817],
   [ 0.3628, -0.4302],
   [ 0.8477, -0.1925],
   [ 1.5683, -1.4538],
   [-0.2494, -1.6042],
   [ 0.0142, -1.6684],
   [-1.6479, -0.3535],
   [ 1.6242, -1.8027]],

  [[-1.2505, -1.3212],
   [ 1.9157, -0.6191],
   [-1.7512,  0.5165],
   [-1.5732, -0.1249],
   [-0.5898, -0.4158],
   [-0.4769, -0.3535],
   [ 0.4879, -0.2919],
   [-0.1908,  1.6932],
   [ 1.0552,  1.9789],
   [ 0.4114, -1.8693]],

  [[-0.6473, -0.2875],
   [ 0.8754,  0.5485],
   [ 1.7395,  0.5469],
   [ 1.9870, -0.5567],
   [-1.5295,  0.1025],
   [-1.3862,  1.2063],
   [-0.7479, -0.0594],
   [ 1.6344, -0.8996],
   [ 0.5685, -0.6133],
   [-1.8893,  1.9618]],

  [[-1.3969, -1.2584],
   [-0.5230,  0.7359],
   [-1.9951,  1.8692],
   [-1.7383,  0.5308],
   [ 0.0171,  1.5884],
   [-0.4782,  1.9707],
   [ 0.3105,  1.5377],
   [-0.3061,  1.2858],
   [-0.0608,  0.7897],
   [-1.5873,  1.1970]],

  [[ 1.3557, -1.8997],
   [ 0.0401,  1.0429],
   [ 0.1588,  0.3099],
   [ 0.8134, -1.7725],
   [-1.4868, -1.9436],
   [-1.3589, -1.8519],
   [ 0.0235, -0.0903],
   [-0.9190, -0.7169],
   [-0.4357, -0.9308],
   [ 1.2138, -0.4648]],

  [[ 0.7425, -1.2723],
   [-1.2882, -0.9645],
   [ 0.9654, -0.3355],
   [ 1.1874,  1.9311],
   [-1.2859, -0.2779],
   [ 0.0146,  0.6645],
   [-1.9226, -1.8894],
   [-1.9399,  0.3529],
   [-0.9138,  1.9704],
   [-1.1756,  0.3996]],

  [[-0.0967, -1.5186],
   [-0.5572,  1.4003],
   [ 1.1635,  0.3844],
   [-1.8407,  1.2133],
   [ 0.3440,  0.8430],
   [ 1.6675, -0.2081],
   [ 1.5182, -0.0991],
   [ 0.4312,  1.1988],
   [ 1.0257,  0.9181],
   [-1.3318,  0.0113]]],


 [[[ 0.5229,  1.7186],
   [ 0.2447, -1.7722],
   [ 1.9558,  0.7897],
   [-0.2866,  1.7106],
   [-1.8656, -0.6073],
   [ 0.2541, -0.5758],
   [ 0.7862, -0.7967],
   [ 0.2434, -0.2500],
   [ 1.9724, -1.6783],
   [-1.2378, -1.8009]],

  [[-0.2075,  1.0700],
   [-1.8012, -0.7954],
   [ 0.7769, -0.3403],
   [ 1.7522, -1.6351],
   [ 1.1443, -0.0109],
   [ 1.1499,  1.1863],
   [ 0.5536, -1.2255],
   [ 1.1178, -1.0296],
   [ 0.5785, -1.0306],
   [ 0.4561,  0.3083]],

  [[ 1.8243, -0.4109],
   [-0.8581, -0.9415],
   [ 1.6291,  1.4080],
   [ 0.2663, -1.4106],
   [-1.3930,  0.4565],
   [ 1.3747, -1.4464],
   [-0.5173,  0.6275],
   [-1.4881, -0.0791],
   [-1.5300, -0.0376],
   [-0.7357,  1.3118]],

  [[ 0.4213, -0.4424],
   [ 0.4718,  0.8544],
   [ 0.6342,  0.5930],
   [-1.4592,  0.9840],
   [-0.2018, -1.5982],
   [ 0.2531,  0.1849],
   [ 0.7382, -1.9418],
   [-0.8996,  0.3930],
   [ 0.3761,  0.0904],
   [-1.3371,  0.9920]],

  [[ 0.8061, -0.4647],
   [-1.5188,  1.0175],
   [ 1.3355, -0.9962],
   [-1.2889, -1.5765],
   [-0.2926,  1.8030],
   [-1.2162,  1.3498],
   [-1.0174, -0.0172],
   [ 1.9736, -0.9195],
   [-0.5556, -0.3043],
   [-0.8305,  0.3111]],

  [[-0.1680,  1.6404],
   [ 0.2090,  0.4003],
   [ 1.7559, -0.1290],
   [-1.2025,  1.6680],
   [-1.0993,  1.4024],
   [-1.6462, -1.7576],
   [ 0.3758, -0.5544],
   [ 1.7620,  0.0187],
   [ 0.7033,  1.0926],
   [ 1.8603,  1.8435]],

  [[-0.2956, -0.1107],
   [-0.9312,  0.3183],
   [-0.9891, -1.5630],
   [-1.4147, -0.0128],
   [-1.5586,  1.3915],
   [ 1.1812, -1.4658],
   [-1.8646,  0.9345],
   [-0.0903, -0.1244],
   [ 0.4643, -0.7193],
   [ 0.2759, -0.7039]],

  [[ 1.6475, -0.4765],
   [ 1.7192,  0.9523],
   [-0.3220, -0.6531],
   [ 0.3992,  1.8709],
   [-1.9119,  1.7273],
   [ 0.7452,  1.1067],
   [-0.1234,  0.8440],
   [ 1.7273, -0.4334],
   [ 0.4186, -0.4803],
   [-1.5332, -0.1467]],

  [[ 1.7229,  0.9690],
   [-1.8442, -1.0074],
   [ 0.4095,  0.1997],
   [ 1.2701, -1.8232],
   [-1.7041, -0.3767],
   [-0.7734, -0.2795],
   [-1.3943, -0.6950],
   [ 0.2118,  1.2398],
   [-1.0847, -0.2509],
   [-0.7644,  1.8773]],

  [[ 0.3199, -1.6492],
   [ 0.4723, -0.6576],
   [ 0.3100, -1.8890],
   [ 1.3837, -0.1127],
   [ 0.4850, -1.1013],
   [-1.5918, -0.1211],
   [ 0.1576, -0.7470],
   [-0.6888, -1.3469],
   [-0.9145,  0.3586],
   [-1.7643,  0.4087]],

  [[-0.7426,  0.7931],
   [-1.1826, -1.7763],
   [ 0.4773, -1.5527],
   [ 1.6368, -0.7878],
   [ 1.8296, -0.6230],
   [ 1.6000, -0.3928],
   [-0.2718, -1.6651],
   [ 1.1806,  0.5856],
   [ 1.9963, -0.4580],
   [-0.0518, -0.0662]],

  [[-1.5275, -0.2008],
   [-1.0342, -1.4198],
   [ 1.5619, -0.8835],
   [ 0.4056,  0.8474],
   [ 0.6425,  1.1953],
   [ 1.0211, -1.4575],
   [ 1.4458,  0.1762],
   [-0.9242, -1.8079],
   [-1.2674, -0.0075],
   [-1.2202, -1.8033]],

  [[ 0.8573,  0.8268],
   [ 1.0164,  1.8607],
   [-0.3526, -0.6622],
   [-0.1924,  0.5389],
   [ 1.6591,  1.8509],
   [ 1.1368,  0.9931],
   [-0.0709,  1.3218],
   [ 1.5008,  0.9046],
   [-1.4040,  1.6321],
   [ 1.0302, -0.2132]]]], device='cuda:0', requires_grad=True), 2, 1, False), kwargs={}, broadcasts_input=False, name=''):


Caused by sample input at index 14: SampleInput(input=Tensor[size=(2, 3, 5, 20), device="cuda:0", dtype=torch.float32], args=(Tensor[size=(2, 13, 10, 2), device="cuda:0", dtype=torch.float32],2,1,False), kwargs={}, broadcasts_input=False, name='')

To execute this test, run the following from the base repo dir:
    PYTORCH_OPINFO_SAMPLE_INPUT_INDEX=14 python test/test_ops_jit.py TestJitCUDA.test_variant_consistency_jit_grid_sampler_2d_cuda_float32

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0
RAW_BUFFERClick to expand / collapse

Platforms: linux

This test was disabled because it is failing in CI. See recent examples and the most recent trunk workflow logs.

Over the past 6 hours, it has been determined flaky in 4 workflow(s) with 11 failures and 4 successes.

Debugging instructions (after clicking on the recent samples link): DO NOT ASSUME THINGS ARE OKAY IF THE CI IS GREEN. We now shield flaky tests from developers so CI will thus be green but it will be harder to parse the logs. To find relevant log snippets:

  1. Click on the workflow logs linked above
  2. Click on the Test step of the job so that it is expanded. Otherwise, the grepping will not work.
  3. Grep for test_variant_consistency_jit_grid_sampler_2d_cuda_float32
  4. There should be several instances run (as flaky tests are rerun in CI) from which you can study the logs.
<details><summary>Sample error message</summary>
Traceback (most recent call last):
  File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 109, in test_variant_consistency_jit
    self.indiv_variant_test_jit(
  File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 157, in indiv_variant_test_jit
    check_against_reference(
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_jit.py", line 140, in check_against_reference
    self.assertEqual(g2, g2_test, atol=5e-4, rtol=1e-4)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 4440, in assertEqual
    raise error_metas.pop()[0].to_error(  # type: ignore[index]
AssertionError: Tensor-likes are not close!

Mismatched elements: 84 / 520 (16.2%)
Greatest absolute difference: 0.005322886630892754 at index (0, 4, 8, 0) (up to 0.0005 allowed)
Greatest relative difference: 11.676852226257324 at index (0, 8, 9, 0) (up to 0.0001 allowed)

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 1175, in test_wrapper
    return test(*args, **kwargs)
  File "/var/lib/jenkins/workspace/test/test_ops_jit.py", line 120, in test_variant_consistency_jit
    raise Exception(variant_error_info) from e  # noqa: TRY002
Exception: 
                 Error testing grid_sampler_2d function variant
                 with dtype: torch.float32
                 with inputs SampleInput(input=tensor([[[[-0.7560, -0.1443, -1.3317,  1.0707, -0.0536,  1.2746, -1.0104,
     1.7234,  0.4455,  0.2850,  0.4191,  1.4160, -1.6165,  1.2780,
    -1.4350,  0.3450, -1.6000,  0.3590,  1.1890,  1.1320],
   [-1.2507, -0.5309, -1.4537,  0.7853,  1.7886, -0.0907,  1.6612,
     1.2864,  0.1156, -0.8906, -0.7386, -1.8012, -1.8938,  0.8986,
     0.0282, -0.9127,  0.3131, -1.2557,  0.7019,  1.8979],
   [-0.9708,  1.1602, -1.4717, -1.8265, -1.3409,  1.7040,  0.1004,
    -0.4273,  0.3805,  0.4131, -1.8698, -0.5582, -1.2652,  0.8626,
     1.7328,  1.7129, -1.0489,  0.3309, -0.0435,  1.1554],
   [ 1.2676, -1.9798, -0.7505, -0.2731, -1.1362,  0.5353, -1.3947,
     1.6763, -1.4323,  0.5500,  0.3345, -0.1932,  0.2958, -1.0673,
     0.1891,  1.8858,  1.9893, -0.4553,  0.4284, -0.4182],
   [ 0.8169, -1.6031, -1.4634,  0.5036, -0.9926,  1.9319, -1.1357,
     1.4299, -0.1737, -0.3567, -0.5193, -0.1565,  1.3256,  0.9212,
    -1.0795,  0.2933, -0.1425, -0.9356, -0.3506, -1.1891]],

  [[-1.4523,  1.6652,  1.8988, -1.5228, -0.4680,  1.7696, -0.0126,
     0.0595, -0.9792,  1.3731, -1.0529,  1.3009, -1.9354, -0.4431,
    -0.3360, -0.8617,  1.6389,  0.5681,  1.4091, -0.4532],
   [ 0.4200,  1.6801, -0.6350, -0.2937,  0.8192, -1.7547, -0.0023,
    -0.8332,  1.7256,  1.5671, -1.9655, -0.9337,  1.1646,  0.1157,
     0.8171, -0.2420,  1.9833, -1.2515, -1.9892,  1.2424],
   [ 1.4888, -0.1588,  1.3372, -1.7930, -0.2623,  0.0159,  1.6651,
    -1.4205, -1.6741,  0.3484, -0.7193,  1.5819,  0.8232,  1.1013,
    -1.7786, -0.8769,  0.7951, -1.1791,  0.6205, -0.4372],
   [ 1.8010, -0.6551, -1.6349, -0.1433, -1.6172,  1.7753, -0.3452,
    -1.8068,  0.2773,  1.0525,  0.2656, -0.0981, -0.6501, -1.4674,
     1.1974,  0.1889, -1.1462,  1.2498,  0.0458, -1.5549],
   [-0.7784, -1.4418,  1.5518, -1.6349,  0.2633, -0.0242,  0.4860,
    -0.2963,  1.9413,  0.8316, -1.9919, -1.5023, -0.7553,  1.7810,
    -0.1282,  1.8717, -0.6003,  1.1921, -1.3241, -0.8015]],

  [[ 0.0387, -0.0683, -1.3119, -0.6655,  1.3270,  0.4466, -0.5344,
     0.1092, -1.6954, -1.7021,  0.2210,  0.0358,  1.7904, -1.5685,
    -0.0929, -0.3973,  0.2937,  0.8790,  0.6636,  1.0969],
   [ 1.1867, -1.8680, -0.7894,  0.1756, -1.8204,  0.7961, -1.3391,
     0.7083, -1.5092,  1.1195, -1.0322, -0.1478, -0.9416, -1.6585,
     1.1229,  0.8414,  1.6447,  0.3158,  1.7880,  0.9119],
   [ 0.8099,  0.3150,  0.6882,  1.6934, -0.2961,  0.3199,  0.9885,
    -0.3853,  1.1251, -0.8356,  1.4918,  1.3761,  1.7323, -1.2555,
    -1.9041,  0.2660, -1.5036, -1.1210,  1.7914,  0.6311],
   [ 1.9929,  1.1131,  1.9899,  1.0328, -1.4152,  1.5587, -0.4122,
    -1.5055,  1.6023, -1.2365, -0.9438,  1.2561, -1.5768,  0.6131,
     1.9208,  1.6280, -1.4219,  1.7101,  0.0306,  0.4187],
   [-1.7767,  1.2516,  0.7330, -1.1223,  0.2094,  0.5165,  1.4007,
     0.6846, -0.1133,  0.6371,  0.6199, -0.0956,  1.5327, -1.6353,
     0.7567,  0.6291, -1.2143, -1.4736,  0.0165,  0.8720]]],


 [[[-1.3789, -1.2527,  1.3401,  0.3406, -0.7042,  1.1389,  0.6666,
    -0.4117, -0.2989, -0.3938,  0.3820, -0.6539,  0.9630,  0.9078,
    -1.2130,  0.3959,  1.1905, -0.8307,  0.9276,  1.8631],
   [ 0.6441, -0.0742, -0.2414,  1.8217,  0.9493,  0.8255, -0.7547,
    -1.2031, -1.3122,  0.5966,  0.7229, -1.6750,  0.0700, -0.6009,
     0.2622,  1.1515, -1.0787, -1.4436, -1.4295,  0.3572],
   [ 0.6079, -0.5907, -0.9739, -0.1204, -1.8401, -0.1114,  0.1417,
     1.0373, -0.0542,  1.5353,  0.0925, -1.4203, -0.6187, -0.3825,
     1.4880,  0.5443,  1.3337, -1.5342,  0.1816,  0.3702],
   [-1.8681,  0.5987, -1.0082, -1.9412, -0.0964, -1.7483,  1.1809,
     0.5175,  1.9990,  1.1500, -1.7001,  0.5029, -0.1091, -1.1460,
     1.7729, -0.5552, -1.4418,  1.4706, -1.9570,  0.4494],
   [ 1.4411,  0.6376, -0.3329, -0.5400, -0.1568,  0.9053, -0.7919,
    -0.5293,  0.5903,  0.0579,  1.8277, -1.6401, -0.8383,  0.5593,
     0.4174, -0.9361, -0.5755, -0.3629,  1.9672,  1.2225]],

  [[-0.5613, -0.3929,  0.2896, -1.7571,  1.9991, -0.4219, -0.7767,
    -1.8055, -0.3581, -1.7616, -1.6126,  0.2505, -0.2944,  0.4794,
    -0.3436,  0.2436,  1.0697, -0.7204,  0.8678,  1.1480],
   [ 1.6177,  0.9862,  0.8313, -1.5676, -0.8029, -0.4710, -0.9152,
     1.3049,  1.1056,  1.7808,  0.6238, -1.4056,  1.3417,  0.1611,
     0.7123, -1.9622, -1.0206,  1.0915,  1.0982,  1.9980],
   [-0.3426,  1.1918,  1.3012, -1.3486, -1.1785,  0.6574,  0.4834,
    -1.5202, -1.4326, -1.0696,  0.7712,  0.6510,  1.4800,  0.4209,
    -1.5200, -1.4426, -0.4412, -0.0307,  0.9065, -1.8860],
   [-0.7202,  0.5170, -0.7852, -0.3009, -0.4691,  1.9814,  1.9398,
    -1.0492,  1.9136, -0.2763, -0.0766, -0.8461, -0.1436,  1.9201,
     1.0925,  0.1738,  0.6860,  0.4168,  0.5323,  0.8805],
   [-0.4933, -1.5355, -1.1386, -1.8716, -0.6593, -0.4551, -1.8895,
    -0.0941,  0.6320, -1.5665,  0.2567, -1.6547,  0.4524,  0.8572,
     0.4672,  0.3772,  0.9701, -0.6117, -0.4405,  0.2073]],

  [[ 0.8025, -0.0595, -0.9438,  0.2717, -0.3834,  0.8637,  0.9916,
    -0.4114,  1.8228, -0.6555, -0.1733, -0.6234,  0.4483,  0.4614,
     0.2109, -0.0100, -0.7163, -0.0803, -0.5773,  1.8567],
   [ 1.1786,  0.0330,  0.9112,  0.1013, -0.1249, -0.5918, -1.8807,
    -0.6012, -0.4238,  1.9679, -1.3820,  1.6903, -1.8332, -0.8515,
     0.0218,  0.7700, -1.7483,  0.7853, -1.9223,  0.1178],
   [-0.6113, -1.2863, -1.8943,  0.6953, -1.7618,  0.8581,  0.9284,
     1.7700, -0.1035, -1.8848,  0.2061, -1.4119, -0.3077,  0.3535,
    -1.3055,  1.3411,  0.9784,  0.2072,  1.7066, -1.2979],
   [-1.7509,  1.1533,  1.9451, -0.6432, -1.2304, -1.1662, -1.2911,
    -0.8734,  0.9899, -1.9781, -0.4201,  1.5957, -1.5980, -0.7309,
    -1.9116, -0.0167, -1.6116, -1.2348,  0.5920,  1.1657],
   [-0.1331,  1.6117,  0.5441, -1.8298, -1.6769,  1.5804, -1.6245,
     1.8640, -1.0085,  1.5367,  0.4123,  0.5132, -0.4380,  1.5536,
    -1.8067,  1.9436, -0.7693, -0.0835,  1.0801,  1.2580]]]],
device='cuda:0', requires_grad=True), args=(tensor([[[[ 0.8840,  0.7284],
   [-1.9067, -1.7008],
   [ 1.8565, -0.3989],
   [-0.8446,  0.7945],
   [-1.1669,  0.7130],
   [-1.8342, -1.8440],
   [-0.0883,  1.5642],
   [ 0.7254,  0.5880],
   [ 0.2307,  1.1799],
   [ 1.4357, -0.6544]],

  [[-1.0316,  1.2807],
   [ 1.0162, -0.9042],
   [ 0.5910,  0.3717],
   [ 0.3884,  0.6074],
   [-1.0467, -1.9530],
   [-0.7783, -0.1390],
   [ 1.5114, -1.0019],
   [-1.1427,  0.9776],
   [-0.7653, -0.7106],
   [ 0.5124,  0.0094]],

  [[-0.2061, -1.0028],
   [ 0.5493,  0.3959],
   [-0.0231,  0.9015],
   [-0.1169,  1.0216],
   [ 1.6647, -0.0388],
   [-1.0319,  0.4510],
   [ 1.5703,  0.4492],
   [-1.9579, -0.3020],
   [ 0.7635,  1.5684],
   [ 0.3821,  1.6603]],

  [[-0.6688,  0.5106],
   [-0.4162,  1.6228],
   [-0.3385,  1.9604],
   [ 0.7010, -1.9517],
   [-1.5050,  0.1919],
   [-1.9553, -1.1527],
   [-0.1650,  0.0051],
   [-0.8922,  1.1805],
   [-0.0818,  1.2613],
   [ 0.7639, -1.4357]],

  [[-1.6299,  0.4255],
   [-0.6200, -0.6759],
   [ 1.0267, -1.6362],
   [ 1.2146, -1.7742],
   [-1.9981, -1.0791],
   [-0.7641,  1.6715],
   [-1.6763, -1.4935],
   [ 1.0545,  1.9053],
   [-1.8099,  0.2377],
   [ 1.8880, -1.7312]],

  [[ 1.1423, -1.4683],
   [ 1.4219,  0.7878],
   [-0.3497, -0.4911],
   [ 0.3866,  0.7407],
   [-1.4957,  0.8446],
   [ 0.3847, -1.4483],
   [-0.3230,  0.6764],
   [-0.1395, -1.7608],
   [ 0.4753, -0.1717],
   [-1.9776, -1.1859]],

  [[ 0.9461,  1.6932],
   [ 1.0734, -0.9379],
   [-0.2483, -1.6817],
   [ 0.3628, -0.4302],
   [ 0.8477, -0.1925],
   [ 1.5683, -1.4538],
   [-0.2494, -1.6042],
   [ 0.0142, -1.6684],
   [-1.6479, -0.3535],
   [ 1.6242, -1.8027]],

  [[-1.2505, -1.3212],
   [ 1.9157, -0.6191],
   [-1.7512,  0.5165],
   [-1.5732, -0.1249],
   [-0.5898, -0.4158],
   [-0.4769, -0.3535],
   [ 0.4879, -0.2919],
   [-0.1908,  1.6932],
   [ 1.0552,  1.9789],
   [ 0.4114, -1.8693]],

  [[-0.6473, -0.2875],
   [ 0.8754,  0.5485],
   [ 1.7395,  0.5469],
   [ 1.9870, -0.5567],
   [-1.5295,  0.1025],
   [-1.3862,  1.2063],
   [-0.7479, -0.0594],
   [ 1.6344, -0.8996],
   [ 0.5685, -0.6133],
   [-1.8893,  1.9618]],

  [[-1.3969, -1.2584],
   [-0.5230,  0.7359],
   [-1.9951,  1.8692],
   [-1.7383,  0.5308],
   [ 0.0171,  1.5884],
   [-0.4782,  1.9707],
   [ 0.3105,  1.5377],
   [-0.3061,  1.2858],
   [-0.0608,  0.7897],
   [-1.5873,  1.1970]],

  [[ 1.3557, -1.8997],
   [ 0.0401,  1.0429],
   [ 0.1588,  0.3099],
   [ 0.8134, -1.7725],
   [-1.4868, -1.9436],
   [-1.3589, -1.8519],
   [ 0.0235, -0.0903],
   [-0.9190, -0.7169],
   [-0.4357, -0.9308],
   [ 1.2138, -0.4648]],

  [[ 0.7425, -1.2723],
   [-1.2882, -0.9645],
   [ 0.9654, -0.3355],
   [ 1.1874,  1.9311],
   [-1.2859, -0.2779],
   [ 0.0146,  0.6645],
   [-1.9226, -1.8894],
   [-1.9399,  0.3529],
   [-0.9138,  1.9704],
   [-1.1756,  0.3996]],

  [[-0.0967, -1.5186],
   [-0.5572,  1.4003],
   [ 1.1635,  0.3844],
   [-1.8407,  1.2133],
   [ 0.3440,  0.8430],
   [ 1.6675, -0.2081],
   [ 1.5182, -0.0991],
   [ 0.4312,  1.1988],
   [ 1.0257,  0.9181],
   [-1.3318,  0.0113]]],


 [[[ 0.5229,  1.7186],
   [ 0.2447, -1.7722],
   [ 1.9558,  0.7897],
   [-0.2866,  1.7106],
   [-1.8656, -0.6073],
   [ 0.2541, -0.5758],
   [ 0.7862, -0.7967],
   [ 0.2434, -0.2500],
   [ 1.9724, -1.6783],
   [-1.2378, -1.8009]],

  [[-0.2075,  1.0700],
   [-1.8012, -0.7954],
   [ 0.7769, -0.3403],
   [ 1.7522, -1.6351],
   [ 1.1443, -0.0109],
   [ 1.1499,  1.1863],
   [ 0.5536, -1.2255],
   [ 1.1178, -1.0296],
   [ 0.5785, -1.0306],
   [ 0.4561,  0.3083]],

  [[ 1.8243, -0.4109],
   [-0.8581, -0.9415],
   [ 1.6291,  1.4080],
   [ 0.2663, -1.4106],
   [-1.3930,  0.4565],
   [ 1.3747, -1.4464],
   [-0.5173,  0.6275],
   [-1.4881, -0.0791],
   [-1.5300, -0.0376],
   [-0.7357,  1.3118]],

  [[ 0.4213, -0.4424],
   [ 0.4718,  0.8544],
   [ 0.6342,  0.5930],
   [-1.4592,  0.9840],
   [-0.2018, -1.5982],
   [ 0.2531,  0.1849],
   [ 0.7382, -1.9418],
   [-0.8996,  0.3930],
   [ 0.3761,  0.0904],
   [-1.3371,  0.9920]],

  [[ 0.8061, -0.4647],
   [-1.5188,  1.0175],
   [ 1.3355, -0.9962],
   [-1.2889, -1.5765],
   [-0.2926,  1.8030],
   [-1.2162,  1.3498],
   [-1.0174, -0.0172],
   [ 1.9736, -0.9195],
   [-0.5556, -0.3043],
   [-0.8305,  0.3111]],

  [[-0.1680,  1.6404],
   [ 0.2090,  0.4003],
   [ 1.7559, -0.1290],
   [-1.2025,  1.6680],
   [-1.0993,  1.4024],
   [-1.6462, -1.7576],
   [ 0.3758, -0.5544],
   [ 1.7620,  0.0187],
   [ 0.7033,  1.0926],
   [ 1.8603,  1.8435]],

  [[-0.2956, -0.1107],
   [-0.9312,  0.3183],
   [-0.9891, -1.5630],
   [-1.4147, -0.0128],
   [-1.5586,  1.3915],
   [ 1.1812, -1.4658],
   [-1.8646,  0.9345],
   [-0.0903, -0.1244],
   [ 0.4643, -0.7193],
   [ 0.2759, -0.7039]],

  [[ 1.6475, -0.4765],
   [ 1.7192,  0.9523],
   [-0.3220, -0.6531],
   [ 0.3992,  1.8709],
   [-1.9119,  1.7273],
   [ 0.7452,  1.1067],
   [-0.1234,  0.8440],
   [ 1.7273, -0.4334],
   [ 0.4186, -0.4803],
   [-1.5332, -0.1467]],

  [[ 1.7229,  0.9690],
   [-1.8442, -1.0074],
   [ 0.4095,  0.1997],
   [ 1.2701, -1.8232],
   [-1.7041, -0.3767],
   [-0.7734, -0.2795],
   [-1.3943, -0.6950],
   [ 0.2118,  1.2398],
   [-1.0847, -0.2509],
   [-0.7644,  1.8773]],

  [[ 0.3199, -1.6492],
   [ 0.4723, -0.6576],
   [ 0.3100, -1.8890],
   [ 1.3837, -0.1127],
   [ 0.4850, -1.1013],
   [-1.5918, -0.1211],
   [ 0.1576, -0.7470],
   [-0.6888, -1.3469],
   [-0.9145,  0.3586],
   [-1.7643,  0.4087]],

  [[-0.7426,  0.7931],
   [-1.1826, -1.7763],
   [ 0.4773, -1.5527],
   [ 1.6368, -0.7878],
   [ 1.8296, -0.6230],
   [ 1.6000, -0.3928],
   [-0.2718, -1.6651],
   [ 1.1806,  0.5856],
   [ 1.9963, -0.4580],
   [-0.0518, -0.0662]],

  [[-1.5275, -0.2008],
   [-1.0342, -1.4198],
   [ 1.5619, -0.8835],
   [ 0.4056,  0.8474],
   [ 0.6425,  1.1953],
   [ 1.0211, -1.4575],
   [ 1.4458,  0.1762],
   [-0.9242, -1.8079],
   [-1.2674, -0.0075],
   [-1.2202, -1.8033]],

  [[ 0.8573,  0.8268],
   [ 1.0164,  1.8607],
   [-0.3526, -0.6622],
   [-0.1924,  0.5389],
   [ 1.6591,  1.8509],
   [ 1.1368,  0.9931],
   [-0.0709,  1.3218],
   [ 1.5008,  0.9046],
   [-1.4040,  1.6321],
   [ 1.0302, -0.2132]]]], device='cuda:0', requires_grad=True), 2, 1, False), kwargs={}, broadcasts_input=False, name=''):


The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 3443, in wrapper
    method(*args, **kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 3443, in wrapper
    method(*args, **kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 439, in instantiated_test
    result = test(self, **param_kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_utils.py", line 1820, in wrapper
    fn(*args, **kwargs)
  File "/opt/conda/envs/py_3.10/lib/python3.10/site-packages/torch/testing/_internal/common_device_type.py", line 1187, in test_wrapper
    raise e_tracked from e
Exception: 
                 Error testing grid_sampler_2d function variant
                 with dtype: torch.float32
                 with inputs SampleInput(input=tensor([[[[-0.7560, -0.1443, -1.3317,  1.0707, -0.0536,  1.2746, -1.0104,
     1.7234,  0.4455,  0.2850,  0.4191,  1.4160, -1.6165,  1.2780,
    -1.4350,  0.3450, -1.6000,  0.3590,  1.1890,  1.1320],
   [-1.2507, -0.5309, -1.4537,  0.7853,  1.7886, -0.0907,  1.6612,
     1.2864,  0.1156, -0.8906, -0.7386, -1.8012, -1.8938,  0.8986,
     0.0282, -0.9127,  0.3131, -1.2557,  0.7019,  1.8979],
   [-0.9708,  1.1602, -1.4717, -1.8265, -1.3409,  1.7040,  0.1004,
    -0.4273,  0.3805,  0.4131, -1.8698, -0.5582, -1.2652,  0.8626,
     1.7328,  1.7129, -1.0489,  0.3309, -0.0435,  1.1554],
   [ 1.2676, -1.9798, -0.7505, -0.2731, -1.1362,  0.5353, -1.3947,
     1.6763, -1.4323,  0.5500,  0.3345, -0.1932,  0.2958, -1.0673,
     0.1891,  1.8858,  1.9893, -0.4553,  0.4284, -0.4182],
   [ 0.8169, -1.6031, -1.4634,  0.5036, -0.9926,  1.9319, -1.1357,
     1.4299, -0.1737, -0.3567, -0.5193, -0.1565,  1.3256,  0.9212,
    -1.0795,  0.2933, -0.1425, -0.9356, -0.3506, -1.1891]],

  [[-1.4523,  1.6652,  1.8988, -1.5228, -0.4680,  1.7696, -0.0126,
     0.0595, -0.9792,  1.3731, -1.0529,  1.3009, -1.9354, -0.4431,
    -0.3360, -0.8617,  1.6389,  0.5681,  1.4091, -0.4532],
   [ 0.4200,  1.6801, -0.6350, -0.2937,  0.8192, -1.7547, -0.0023,
    -0.8332,  1.7256,  1.5671, -1.9655, -0.9337,  1.1646,  0.1157,
     0.8171, -0.2420,  1.9833, -1.2515, -1.9892,  1.2424],
   [ 1.4888, -0.1588,  1.3372, -1.7930, -0.2623,  0.0159,  1.6651,
    -1.4205, -1.6741,  0.3484, -0.7193,  1.5819,  0.8232,  1.1013,
    -1.7786, -0.8769,  0.7951, -1.1791,  0.6205, -0.4372],
   [ 1.8010, -0.6551, -1.6349, -0.1433, -1.6172,  1.7753, -0.3452,
    -1.8068,  0.2773,  1.0525,  0.2656, -0.0981, -0.6501, -1.4674,
     1.1974,  0.1889, -1.1462,  1.2498,  0.0458, -1.5549],
   [-0.7784, -1.4418,  1.5518, -1.6349,  0.2633, -0.0242,  0.4860,
    -0.2963,  1.9413,  0.8316, -1.9919, -1.5023, -0.7553,  1.7810,
    -0.1282,  1.8717, -0.6003,  1.1921, -1.3241, -0.8015]],

  [[ 0.0387, -0.0683, -1.3119, -0.6655,  1.3270,  0.4466, -0.5344,
     0.1092, -1.6954, -1.7021,  0.2210,  0.0358,  1.7904, -1.5685,
    -0.0929, -0.3973,  0.2937,  0.8790,  0.6636,  1.0969],
   [ 1.1867, -1.8680, -0.7894,  0.1756, -1.8204,  0.7961, -1.3391,
     0.7083, -1.5092,  1.1195, -1.0322, -0.1478, -0.9416, -1.6585,
     1.1229,  0.8414,  1.6447,  0.3158,  1.7880,  0.9119],
   [ 0.8099,  0.3150,  0.6882,  1.6934, -0.2961,  0.3199,  0.9885,
    -0.3853,  1.1251, -0.8356,  1.4918,  1.3761,  1.7323, -1.2555,
    -1.9041,  0.2660, -1.5036, -1.1210,  1.7914,  0.6311],
   [ 1.9929,  1.1131,  1.9899,  1.0328, -1.4152,  1.5587, -0.4122,
    -1.5055,  1.6023, -1.2365, -0.9438,  1.2561, -1.5768,  0.6131,
     1.9208,  1.6280, -1.4219,  1.7101,  0.0306,  0.4187],
   [-1.7767,  1.2516,  0.7330, -1.1223,  0.2094,  0.5165,  1.4007,
     0.6846, -0.1133,  0.6371,  0.6199, -0.0956,  1.5327, -1.6353,
     0.7567,  0.6291, -1.2143, -1.4736,  0.0165,  0.8720]]],


 [[[-1.3789, -1.2527,  1.3401,  0.3406, -0.7042,  1.1389,  0.6666,
    -0.4117, -0.2989, -0.3938,  0.3820, -0.6539,  0.9630,  0.9078,
    -1.2130,  0.3959,  1.1905, -0.8307,  0.9276,  1.8631],
   [ 0.6441, -0.0742, -0.2414,  1.8217,  0.9493,  0.8255, -0.7547,
    -1.2031, -1.3122,  0.5966,  0.7229, -1.6750,  0.0700, -0.6009,
     0.2622,  1.1515, -1.0787, -1.4436, -1.4295,  0.3572],
   [ 0.6079, -0.5907, -0.9739, -0.1204, -1.8401, -0.1114,  0.1417,
     1.0373, -0.0542,  1.5353,  0.0925, -1.4203, -0.6187, -0.3825,
     1.4880,  0.5443,  1.3337, -1.5342,  0.1816,  0.3702],
   [-1.8681,  0.5987, -1.0082, -1.9412, -0.0964, -1.7483,  1.1809,
     0.5175,  1.9990,  1.1500, -1.7001,  0.5029, -0.1091, -1.1460,
     1.7729, -0.5552, -1.4418,  1.4706, -1.9570,  0.4494],
   [ 1.4411,  0.6376, -0.3329, -0.5400, -0.1568,  0.9053, -0.7919,
    -0.5293,  0.5903,  0.0579,  1.8277, -1.6401, -0.8383,  0.5593,
     0.4174, -0.9361, -0.5755, -0.3629,  1.9672,  1.2225]],

  [[-0.5613, -0.3929,  0.2896, -1.7571,  1.9991, -0.4219, -0.7767,
    -1.8055, -0.3581, -1.7616, -1.6126,  0.2505, -0.2944,  0.4794,
    -0.3436,  0.2436,  1.0697, -0.7204,  0.8678,  1.1480],
   [ 1.6177,  0.9862,  0.8313, -1.5676, -0.8029, -0.4710, -0.9152,
     1.3049,  1.1056,  1.7808,  0.6238, -1.4056,  1.3417,  0.1611,
     0.7123, -1.9622, -1.0206,  1.0915,  1.0982,  1.9980],
   [-0.3426,  1.1918,  1.3012, -1.3486, -1.1785,  0.6574,  0.4834,
    -1.5202, -1.4326, -1.0696,  0.7712,  0.6510,  1.4800,  0.4209,
    -1.5200, -1.4426, -0.4412, -0.0307,  0.9065, -1.8860],
   [-0.7202,  0.5170, -0.7852, -0.3009, -0.4691,  1.9814,  1.9398,
    -1.0492,  1.9136, -0.2763, -0.0766, -0.8461, -0.1436,  1.9201,
     1.0925,  0.1738,  0.6860,  0.4168,  0.5323,  0.8805],
   [-0.4933, -1.5355, -1.1386, -1.8716, -0.6593, -0.4551, -1.8895,
    -0.0941,  0.6320, -1.5665,  0.2567, -1.6547,  0.4524,  0.8572,
     0.4672,  0.3772,  0.9701, -0.6117, -0.4405,  0.2073]],

  [[ 0.8025, -0.0595, -0.9438,  0.2717, -0.3834,  0.8637,  0.9916,
    -0.4114,  1.8228, -0.6555, -0.1733, -0.6234,  0.4483,  0.4614,
     0.2109, -0.0100, -0.7163, -0.0803, -0.5773,  1.8567],
   [ 1.1786,  0.0330,  0.9112,  0.1013, -0.1249, -0.5918, -1.8807,
    -0.6012, -0.4238,  1.9679, -1.3820,  1.6903, -1.8332, -0.8515,
     0.0218,  0.7700, -1.7483,  0.7853, -1.9223,  0.1178],
   [-0.6113, -1.2863, -1.8943,  0.6953, -1.7618,  0.8581,  0.9284,
     1.7700, -0.1035, -1.8848,  0.2061, -1.4119, -0.3077,  0.3535,
    -1.3055,  1.3411,  0.9784,  0.2072,  1.7066, -1.2979],
   [-1.7509,  1.1533,  1.9451, -0.6432, -1.2304, -1.1662, -1.2911,
    -0.8734,  0.9899, -1.9781, -0.4201,  1.5957, -1.5980, -0.7309,
    -1.9116, -0.0167, -1.6116, -1.2348,  0.5920,  1.1657],
   [-0.1331,  1.6117,  0.5441, -1.8298, -1.6769,  1.5804, -1.6245,
     1.8640, -1.0085,  1.5367,  0.4123,  0.5132, -0.4380,  1.5536,
    -1.8067,  1.9436, -0.7693, -0.0835,  1.0801,  1.2580]]]],
device='cuda:0', requires_grad=True), args=(tensor([[[[ 0.8840,  0.7284],
   [-1.9067, -1.7008],
   [ 1.8565, -0.3989],
   [-0.8446,  0.7945],
   [-1.1669,  0.7130],
   [-1.8342, -1.8440],
   [-0.0883,  1.5642],
   [ 0.7254,  0.5880],
   [ 0.2307,  1.1799],
   [ 1.4357, -0.6544]],

  [[-1.0316,  1.2807],
   [ 1.0162, -0.9042],
   [ 0.5910,  0.3717],
   [ 0.3884,  0.6074],
   [-1.0467, -1.9530],
   [-0.7783, -0.1390],
   [ 1.5114, -1.0019],
   [-1.1427,  0.9776],
   [-0.7653, -0.7106],
   [ 0.5124,  0.0094]],

  [[-0.2061, -1.0028],
   [ 0.5493,  0.3959],
   [-0.0231,  0.9015],
   [-0.1169,  1.0216],
   [ 1.6647, -0.0388],
   [-1.0319,  0.4510],
   [ 1.5703,  0.4492],
   [-1.9579, -0.3020],
   [ 0.7635,  1.5684],
   [ 0.3821,  1.6603]],

  [[-0.6688,  0.5106],
   [-0.4162,  1.6228],
   [-0.3385,  1.9604],
   [ 0.7010, -1.9517],
   [-1.5050,  0.1919],
   [-1.9553, -1.1527],
   [-0.1650,  0.0051],
   [-0.8922,  1.1805],
   [-0.0818,  1.2613],
   [ 0.7639, -1.4357]],

  [[-1.6299,  0.4255],
   [-0.6200, -0.6759],
   [ 1.0267, -1.6362],
   [ 1.2146, -1.7742],
   [-1.9981, -1.0791],
   [-0.7641,  1.6715],
   [-1.6763, -1.4935],
   [ 1.0545,  1.9053],
   [-1.8099,  0.2377],
   [ 1.8880, -1.7312]],

  [[ 1.1423, -1.4683],
   [ 1.4219,  0.7878],
   [-0.3497, -0.4911],
   [ 0.3866,  0.7407],
   [-1.4957,  0.8446],
   [ 0.3847, -1.4483],
   [-0.3230,  0.6764],
   [-0.1395, -1.7608],
   [ 0.4753, -0.1717],
   [-1.9776, -1.1859]],

  [[ 0.9461,  1.6932],
   [ 1.0734, -0.9379],
   [-0.2483, -1.6817],
   [ 0.3628, -0.4302],
   [ 0.8477, -0.1925],
   [ 1.5683, -1.4538],
   [-0.2494, -1.6042],
   [ 0.0142, -1.6684],
   [-1.6479, -0.3535],
   [ 1.6242, -1.8027]],

  [[-1.2505, -1.3212],
   [ 1.9157, -0.6191],
   [-1.7512,  0.5165],
   [-1.5732, -0.1249],
   [-0.5898, -0.4158],
   [-0.4769, -0.3535],
   [ 0.4879, -0.2919],
   [-0.1908,  1.6932],
   [ 1.0552,  1.9789],
   [ 0.4114, -1.8693]],

  [[-0.6473, -0.2875],
   [ 0.8754,  0.5485],
   [ 1.7395,  0.5469],
   [ 1.9870, -0.5567],
   [-1.5295,  0.1025],
   [-1.3862,  1.2063],
   [-0.7479, -0.0594],
   [ 1.6344, -0.8996],
   [ 0.5685, -0.6133],
   [-1.8893,  1.9618]],

  [[-1.3969, -1.2584],
   [-0.5230,  0.7359],
   [-1.9951,  1.8692],
   [-1.7383,  0.5308],
   [ 0.0171,  1.5884],
   [-0.4782,  1.9707],
   [ 0.3105,  1.5377],
   [-0.3061,  1.2858],
   [-0.0608,  0.7897],
   [-1.5873,  1.1970]],

  [[ 1.3557, -1.8997],
   [ 0.0401,  1.0429],
   [ 0.1588,  0.3099],
   [ 0.8134, -1.7725],
   [-1.4868, -1.9436],
   [-1.3589, -1.8519],
   [ 0.0235, -0.0903],
   [-0.9190, -0.7169],
   [-0.4357, -0.9308],
   [ 1.2138, -0.4648]],

  [[ 0.7425, -1.2723],
   [-1.2882, -0.9645],
   [ 0.9654, -0.3355],
   [ 1.1874,  1.9311],
   [-1.2859, -0.2779],
   [ 0.0146,  0.6645],
   [-1.9226, -1.8894],
   [-1.9399,  0.3529],
   [-0.9138,  1.9704],
   [-1.1756,  0.3996]],

  [[-0.0967, -1.5186],
   [-0.5572,  1.4003],
   [ 1.1635,  0.3844],
   [-1.8407,  1.2133],
   [ 0.3440,  0.8430],
   [ 1.6675, -0.2081],
   [ 1.5182, -0.0991],
   [ 0.4312,  1.1988],
   [ 1.0257,  0.9181],
   [-1.3318,  0.0113]]],


 [[[ 0.5229,  1.7186],
   [ 0.2447, -1.7722],
   [ 1.9558,  0.7897],
   [-0.2866,  1.7106],
   [-1.8656, -0.6073],
   [ 0.2541, -0.5758],
   [ 0.7862, -0.7967],
   [ 0.2434, -0.2500],
   [ 1.9724, -1.6783],
   [-1.2378, -1.8009]],

  [[-0.2075,  1.0700],
   [-1.8012, -0.7954],
   [ 0.7769, -0.3403],
   [ 1.7522, -1.6351],
   [ 1.1443, -0.0109],
   [ 1.1499,  1.1863],
   [ 0.5536, -1.2255],
   [ 1.1178, -1.0296],
   [ 0.5785, -1.0306],
   [ 0.4561,  0.3083]],

  [[ 1.8243, -0.4109],
   [-0.8581, -0.9415],
   [ 1.6291,  1.4080],
   [ 0.2663, -1.4106],
   [-1.3930,  0.4565],
   [ 1.3747, -1.4464],
   [-0.5173,  0.6275],
   [-1.4881, -0.0791],
   [-1.5300, -0.0376],
   [-0.7357,  1.3118]],

  [[ 0.4213, -0.4424],
   [ 0.4718,  0.8544],
   [ 0.6342,  0.5930],
   [-1.4592,  0.9840],
   [-0.2018, -1.5982],
   [ 0.2531,  0.1849],
   [ 0.7382, -1.9418],
   [-0.8996,  0.3930],
   [ 0.3761,  0.0904],
   [-1.3371,  0.9920]],

  [[ 0.8061, -0.4647],
   [-1.5188,  1.0175],
   [ 1.3355, -0.9962],
   [-1.2889, -1.5765],
   [-0.2926,  1.8030],
   [-1.2162,  1.3498],
   [-1.0174, -0.0172],
   [ 1.9736, -0.9195],
   [-0.5556, -0.3043],
   [-0.8305,  0.3111]],

  [[-0.1680,  1.6404],
   [ 0.2090,  0.4003],
   [ 1.7559, -0.1290],
   [-1.2025,  1.6680],
   [-1.0993,  1.4024],
   [-1.6462, -1.7576],
   [ 0.3758, -0.5544],
   [ 1.7620,  0.0187],
   [ 0.7033,  1.0926],
   [ 1.8603,  1.8435]],

  [[-0.2956, -0.1107],
   [-0.9312,  0.3183],
   [-0.9891, -1.5630],
   [-1.4147, -0.0128],
   [-1.5586,  1.3915],
   [ 1.1812, -1.4658],
   [-1.8646,  0.9345],
   [-0.0903, -0.1244],
   [ 0.4643, -0.7193],
   [ 0.2759, -0.7039]],

  [[ 1.6475, -0.4765],
   [ 1.7192,  0.9523],
   [-0.3220, -0.6531],
   [ 0.3992,  1.8709],
   [-1.9119,  1.7273],
   [ 0.7452,  1.1067],
   [-0.1234,  0.8440],
   [ 1.7273, -0.4334],
   [ 0.4186, -0.4803],
   [-1.5332, -0.1467]],

  [[ 1.7229,  0.9690],
   [-1.8442, -1.0074],
   [ 0.4095,  0.1997],
   [ 1.2701, -1.8232],
   [-1.7041, -0.3767],
   [-0.7734, -0.2795],
   [-1.3943, -0.6950],
   [ 0.2118,  1.2398],
   [-1.0847, -0.2509],
   [-0.7644,  1.8773]],

  [[ 0.3199, -1.6492],
   [ 0.4723, -0.6576],
   [ 0.3100, -1.8890],
   [ 1.3837, -0.1127],
   [ 0.4850, -1.1013],
   [-1.5918, -0.1211],
   [ 0.1576, -0.7470],
   [-0.6888, -1.3469],
   [-0.9145,  0.3586],
   [-1.7643,  0.4087]],

  [[-0.7426,  0.7931],
   [-1.1826, -1.7763],
   [ 0.4773, -1.5527],
   [ 1.6368, -0.7878],
   [ 1.8296, -0.6230],
   [ 1.6000, -0.3928],
   [-0.2718, -1.6651],
   [ 1.1806,  0.5856],
   [ 1.9963, -0.4580],
   [-0.0518, -0.0662]],

  [[-1.5275, -0.2008],
   [-1.0342, -1.4198],
   [ 1.5619, -0.8835],
   [ 0.4056,  0.8474],
   [ 0.6425,  1.1953],
   [ 1.0211, -1.4575],
   [ 1.4458,  0.1762],
   [-0.9242, -1.8079],
   [-1.2674, -0.0075],
   [-1.2202, -1.8033]],

  [[ 0.8573,  0.8268],
   [ 1.0164,  1.8607],
   [-0.3526, -0.6622],
   [-0.1924,  0.5389],
   [ 1.6591,  1.8509],
   [ 1.1368,  0.9931],
   [-0.0709,  1.3218],
   [ 1.5008,  0.9046],
   [-1.4040,  1.6321],
   [ 1.0302, -0.2132]]]], device='cuda:0', requires_grad=True), 2, 1, False), kwargs={}, broadcasts_input=False, name=''):


Caused by sample input at index 14: SampleInput(input=Tensor[size=(2, 3, 5, 20), device="cuda:0", dtype=torch.float32], args=(Tensor[size=(2, 13, 10, 2), device="cuda:0", dtype=torch.float32],2,1,False), kwargs={}, broadcasts_input=False, name='')

To execute this test, run the following from the base repo dir:
    PYTORCH_OPINFO_SAMPLE_INPUT_INDEX=14 python test/test_ops_jit.py TestJitCUDA.test_variant_consistency_jit_grid_sampler_2d_cuda_float32

This message can be suppressed by setting PYTORCH_PRINT_REPRO_ON_FAILURE=0
</details>

Test file path: test_ops_jit.py

For all disabled tests (by GitHub issue), see https://hud.pytorch.org/disabled.

extent analysis

TL;DR

The test test_variant_consistency_jit_grid_sampler_2d_cuda_float32 is failing due to a numerical inconsistency in the grid_sampler_2d function on CUDA devices.

Guidance

  1. Numerical Inconsistency: The error message indicates a numerical inconsistency between the expected and actual outputs of the grid_sampler_2d function.
  2. Device-Specific Issue: The issue is specific to CUDA devices, as indicated by the device='cuda:0' parameter in the error message.
  3. Input-Specific Failure: The test fails for a specific input at index 14, which can be reproduced by running the test with the environment variable PYTORCH_OPINFO_SAMPLE_INPUT_INDEX=14.
  4. Reproduction Command: To reproduce the failure, run the command PYTORCH_OPINFO_SAMPLE_INPUT_INDEX=14 python test/test_ops_jit.py TestJitCUDA.test_variant_consistency_jit_grid_sampler_2d_cuda_float32 from the base repository directory.

Example

No code example is provided, as the issue is related to a specific test case and input.

Notes

  • The issue may be related to numerical instability or rounding errors in the grid_sampler_2d function on CUDA devices.
  • The test failure is specific to the float32 data type, which may indicate a precision-related issue.

Recommendation

Apply a workaround by increasing the tolerance for numerical comparisons in the test_variant_consistency_jit_grid_sampler_2d_cuda_float32 test, or investigate the numerical instability issue in the grid_sampler_2d function on CUDA devices.

Vote matrix · Quick signals

Works
Did the solution work? Tap to confirm.
Easy Fix
Was it a quick fix?
Time Saver
Did it save you time?
Blocking
Was it severely blocking?
Common Issue
Are others likely hitting this too?
Flaky / Intermittent
Is it intermittent?
Verified / Reproducible
Can you reproduce it reliably?
Loading…

Still need to ship something?

×6

Another batch ranked right after the header list — different links, same matching logic.

Back to top recommendations

TRENDING