From 69d4f474d8c1cc9df5337b5eee006367c1ff7069 Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Thu, 25 Jul 2024 11:21:22 +0300 Subject: [PATCH] amd resnet pf (#5703) --- test/test_linearizer_failures.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/test/test_linearizer_failures.py b/test/test_linearizer_failures.py index 1e7d5948..e639a506 100644 --- a/test/test_linearizer_failures.py +++ b/test/test_linearizer_failures.py @@ -341,5 +341,16 @@ class TestLinearizerFailures(unittest.TestCase): helper_test_lin(Kernel(ast), opts=opts, failed_platforms=[]) # END METAL=1 ./examples/beautiful_mnist.py failures + @unittest.skipIf(CI, "for real AMD GPU") + def test_failure_41(self): + # One more resnet crash with a page fault on AMD. + ast = LazyOp(MetaOps.KERNEL, arg=None, src=(LazyOp(BufferOps.STORE, arg=MemBuffer(idx=0, dtype=dtypes.half, st=ShapeTracker(views=(View(shape=(256, 1, 128, 28, 28, 1, 1, 1), strides=(100352, 0, 784, 28, 1, 0, 0, 0), offset=0, mask=None, contiguous=True),))), src=(LazyOp(UnaryOps.CAST, arg=dtypes.half, src=( + LazyOp(ReduceOps.SUM, arg=(5, 6, 7), src=( + LazyOp(UnaryOps.CAST, arg=dtypes.float, src=( + LazyOp(BinaryOps.MUL, arg=None, src=( + LazyOp(BufferOps.LOAD, arg=MemBuffer(idx=1, dtype=dtypes.half, st=ShapeTracker(views=(View(shape=(1, 256, 1, 128, 4, 58, 4, 58), strides=(0, 401408, 0, 3136, 0, 56, 0, 1), offset=-57, mask=((0, 1), (0, 256), (0, 1), (0, 128), (0, 4), (1, 57), (0, 4), (1, 57)), contiguous=False), View(shape=(256, 1, 128, 28, 28, 128, 3, 3), strides=(6889472, 0, 0, 464, 2, 53824, 13688, 59), offset=0, mask=None, contiguous=False)))), src=()), LazyOp(BufferOps.LOAD, arg=MemBuffer(idx=2, dtype=dtypes.half, st=ShapeTracker(views=(View(shape=(256, 1, 128, 28, 28, 128, 3, 3), strides=(0, 0, 1152, 0, 0, 9, 3, 1), offset=0, mask=None, contiguous=False),))), src=()),)),)),)),)),)),)) + opts=[Opt(op=OptOps.TC, axis=5, amt=2), Opt(op=OptOps.UNROLL, axis=0, amt=0)] + helper_test_lin(Kernel(ast), opts=opts, failed_platforms=["AMD", "HIP"]) + if __name__ == '__main__': unittest.main()