import torch
from torch._inductor.select_algorithm import extern_kernels
import triton
import triton.language as tl
from torch._inductor.runtime.triton_heuristics import grid
from torch._C import _cuda_getCurrentRawStream as get_raw_stream
from torch._inductor.runtime import triton_helpers
import torch.nn as nn
from math import sqrt
assert_size_stride = torch._C._dynamo.guards.assert_size_stride
empty_strided_cuda = torch._C._dynamo.guards._empty_strided_cuda


@triton.jit
def triton_poi_fused_relu_0(in_ptr0, out_ptr0, xnumel, XBLOCK: tl.constexpr):
    xoffset = tl.program_id(0) * XBLOCK
    xindex = xoffset + tl.arange(0, XBLOCK)[:]
    tl.full([XBLOCK], True, tl.int1)
    x0 = xindex
    tmp0 = tl.load(in_ptr0 + x0, None)
    tmp1 = tl.full([1], 0, tl.int32)
    tmp2 = triton_helpers.maximum(tmp1, tmp0)
    tl.store(out_ptr0 + x0, tmp2, None)


@triton.jit
def triton_poi_fused_relu_1(in_out_ptr0, xnumel, XBLOCK: tl.constexpr):
    xoffset = tl.program_id(0) * XBLOCK
    xindex = xoffset + tl.arange(0, XBLOCK)[:]
    tl.full([XBLOCK], True, tl.int1)
    x0 = xindex
    tmp0 = tl.load(in_out_ptr0 + x0, None)
    tmp1 = tl.full([1], 0, tl.int32)
    tmp2 = triton_helpers.maximum(tmp1, tmp0)
    tl.store(in_out_ptr0 + x0, tmp2, None)


@triton.jit
def triton_poi_fused_add_relu_2(in_out_ptr0, in_ptr0, xnumel, XBLOCK: tl.
    constexpr):
    xoffset = tl.program_id(0) * XBLOCK
    xindex = xoffset + tl.arange(0, XBLOCK)[:]
    tl.full([XBLOCK], True, tl.int1)
    x0 = xindex
    tmp0 = tl.load(in_out_ptr0 + x0, None)
    tmp1 = tl.load(in_ptr0 + x0, None)
    tmp2 = tmp0 + tmp1
    tmp3 = tl.full([1], 0, tl.int32)
    tmp4 = triton_helpers.maximum(tmp3, tmp2)
    tl.store(in_out_ptr0 + x0, tmp4, None)


@triton.jit
def triton_poi_fused_add_3(in_out_ptr0, in_ptr0, out_ptr0, xnumel, XBLOCK:
    tl.constexpr):
    xoffset = tl.program_id(0) * XBLOCK
    xindex = xoffset + tl.arange(0, XBLOCK)[:]
    tl.full([XBLOCK], True, tl.int1)
    x0 = xindex
    tmp0 = tl.load(in_out_ptr0 + x0, None)
    tmp1 = tl.load(in_ptr0 + x0, None)
    tmp2 = tmp0 + tmp1
    tl.store(in_out_ptr0 + x0, tmp2, None)
    tl.store(out_ptr0 + x0, tmp1, None)


def call(args):
    primals_1, primals_2, primals_3, primals_4, primals_5 = args
    args.clear()
    assert_size_stride(primals_1, (4, 1, 64, 64), (4096, 4096, 64, 1))
    assert_size_stride(primals_2, (128, 1, 3, 3), (9, 9, 3, 1))
    assert_size_stride(primals_3, (128, 128, 3, 3), (1152, 9, 3, 1))
    assert_size_stride(primals_4, (128, 128, 3, 3), (1152, 9, 3, 1))
    assert_size_stride(primals_5, (1, 128, 3, 3), (1152, 9, 3, 1))
    with torch.cuda._DeviceGuard(0):
        torch.cuda.set_device(0)
        buf0 = empty_strided_cuda((4, 1, 64, 64), (4096, 4096, 64, 1),
            torch.float32)
        get_raw_stream(0)
        triton_poi_fused_relu_0[grid(16384)](primals_1, buf0, 16384, XBLOCK
            =256, num_warps=4, num_stages=1)
        buf1 = extern_kernels.convolution(buf0, primals_2, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf1, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf2 = buf1
        del buf1
        triton_poi_fused_relu_1[grid(2097152)](buf2, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf3 = extern_kernels.convolution(buf2, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf3, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf4 = buf3
        del buf3
        triton_poi_fused_relu_1[grid(2097152)](buf4, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf5 = extern_kernels.convolution(buf4, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf5, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf6 = buf5
        del buf5
        triton_poi_fused_add_relu_2[grid(2097152)](buf6, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf7 = extern_kernels.convolution(buf6, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf7, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf8 = buf7
        del buf7
        triton_poi_fused_relu_1[grid(2097152)](buf8, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf9 = extern_kernels.convolution(buf8, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf9, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf10 = buf9
        del buf9
        triton_poi_fused_add_relu_2[grid(2097152)](buf10, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf11 = extern_kernels.convolution(buf10, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf11, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf12 = buf11
        del buf11
        triton_poi_fused_relu_1[grid(2097152)](buf12, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf13 = extern_kernels.convolution(buf12, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf13, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf14 = buf13
        del buf13
        triton_poi_fused_add_relu_2[grid(2097152)](buf14, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf15 = extern_kernels.convolution(buf14, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf15, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf16 = buf15
        del buf15
        triton_poi_fused_relu_1[grid(2097152)](buf16, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf17 = extern_kernels.convolution(buf16, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf17, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf18 = buf17
        del buf17
        triton_poi_fused_add_relu_2[grid(2097152)](buf18, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf19 = extern_kernels.convolution(buf18, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf19, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf20 = buf19
        del buf19
        triton_poi_fused_relu_1[grid(2097152)](buf20, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf21 = extern_kernels.convolution(buf20, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf21, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf22 = buf21
        del buf21
        triton_poi_fused_add_relu_2[grid(2097152)](buf22, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf23 = extern_kernels.convolution(buf22, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf23, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf24 = buf23
        del buf23
        triton_poi_fused_relu_1[grid(2097152)](buf24, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf25 = extern_kernels.convolution(buf24, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf25, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf26 = buf25
        del buf25
        triton_poi_fused_add_relu_2[grid(2097152)](buf26, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf27 = extern_kernels.convolution(buf26, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf27, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf28 = buf27
        del buf27
        triton_poi_fused_relu_1[grid(2097152)](buf28, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf29 = extern_kernels.convolution(buf28, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf29, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf30 = buf29
        del buf29
        triton_poi_fused_add_relu_2[grid(2097152)](buf30, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf31 = extern_kernels.convolution(buf30, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf31, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf32 = buf31
        del buf31
        triton_poi_fused_relu_1[grid(2097152)](buf32, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf33 = extern_kernels.convolution(buf32, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf33, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf34 = buf33
        del buf33
        triton_poi_fused_add_relu_2[grid(2097152)](buf34, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf35 = extern_kernels.convolution(buf34, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf35, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf36 = buf35
        del buf35
        triton_poi_fused_relu_1[grid(2097152)](buf36, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf37 = extern_kernels.convolution(buf36, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf37, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf38 = buf37
        del buf37
        triton_poi_fused_add_relu_2[grid(2097152)](buf38, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf39 = extern_kernels.convolution(buf38, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf39, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf40 = buf39
        del buf39
        triton_poi_fused_relu_1[grid(2097152)](buf40, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf41 = extern_kernels.convolution(buf40, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf41, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf42 = buf41
        del buf41
        triton_poi_fused_add_relu_2[grid(2097152)](buf42, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf43 = extern_kernels.convolution(buf42, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf43, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf44 = buf43
        del buf43
        triton_poi_fused_relu_1[grid(2097152)](buf44, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf45 = extern_kernels.convolution(buf44, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf45, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf46 = buf45
        del buf45
        triton_poi_fused_add_relu_2[grid(2097152)](buf46, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf47 = extern_kernels.convolution(buf46, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf47, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf48 = buf47
        del buf47
        triton_poi_fused_relu_1[grid(2097152)](buf48, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf49 = extern_kernels.convolution(buf48, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf49, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf50 = buf49
        del buf49
        triton_poi_fused_add_relu_2[grid(2097152)](buf50, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf51 = extern_kernels.convolution(buf50, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf51, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf52 = buf51
        del buf51
        triton_poi_fused_relu_1[grid(2097152)](buf52, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf53 = extern_kernels.convolution(buf52, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf53, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf54 = buf53
        del buf53
        triton_poi_fused_add_relu_2[grid(2097152)](buf54, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf55 = extern_kernels.convolution(buf54, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf55, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf56 = buf55
        del buf55
        triton_poi_fused_relu_1[grid(2097152)](buf56, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf57 = extern_kernels.convolution(buf56, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf57, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf58 = buf57
        del buf57
        triton_poi_fused_add_relu_2[grid(2097152)](buf58, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf59 = extern_kernels.convolution(buf58, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf59, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf60 = buf59
        del buf59
        triton_poi_fused_relu_1[grid(2097152)](buf60, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf61 = extern_kernels.convolution(buf60, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf61, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf62 = buf61
        del buf61
        triton_poi_fused_add_relu_2[grid(2097152)](buf62, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf63 = extern_kernels.convolution(buf62, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf63, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf64 = buf63
        del buf63
        triton_poi_fused_relu_1[grid(2097152)](buf64, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf65 = extern_kernels.convolution(buf64, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf65, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf66 = buf65
        del buf65
        triton_poi_fused_add_relu_2[grid(2097152)](buf66, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf67 = extern_kernels.convolution(buf66, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf67, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf68 = buf67
        del buf67
        triton_poi_fused_relu_1[grid(2097152)](buf68, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf69 = extern_kernels.convolution(buf68, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf69, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf70 = buf69
        del buf69
        triton_poi_fused_add_relu_2[grid(2097152)](buf70, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf71 = extern_kernels.convolution(buf70, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf71, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf72 = buf71
        del buf71
        triton_poi_fused_relu_1[grid(2097152)](buf72, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf73 = extern_kernels.convolution(buf72, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf73, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf74 = buf73
        del buf73
        triton_poi_fused_add_relu_2[grid(2097152)](buf74, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf75 = extern_kernels.convolution(buf74, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf75, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf76 = buf75
        del buf75
        triton_poi_fused_relu_1[grid(2097152)](buf76, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf77 = extern_kernels.convolution(buf76, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf77, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf78 = buf77
        del buf77
        triton_poi_fused_add_relu_2[grid(2097152)](buf78, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf79 = extern_kernels.convolution(buf78, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf79, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf80 = buf79
        del buf79
        triton_poi_fused_relu_1[grid(2097152)](buf80, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf81 = extern_kernels.convolution(buf80, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf81, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf82 = buf81
        del buf81
        triton_poi_fused_add_relu_2[grid(2097152)](buf82, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf83 = extern_kernels.convolution(buf82, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf83, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf84 = buf83
        del buf83
        triton_poi_fused_relu_1[grid(2097152)](buf84, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf85 = extern_kernels.convolution(buf84, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf85, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf86 = buf85
        del buf85
        triton_poi_fused_add_relu_2[grid(2097152)](buf86, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf87 = extern_kernels.convolution(buf86, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf87, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf88 = buf87
        del buf87
        triton_poi_fused_relu_1[grid(2097152)](buf88, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf89 = extern_kernels.convolution(buf88, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf89, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf90 = buf89
        del buf89
        triton_poi_fused_add_relu_2[grid(2097152)](buf90, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf91 = extern_kernels.convolution(buf90, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf91, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf92 = buf91
        del buf91
        triton_poi_fused_relu_1[grid(2097152)](buf92, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf93 = extern_kernels.convolution(buf92, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf93, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf94 = buf93
        del buf93
        triton_poi_fused_add_relu_2[grid(2097152)](buf94, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf95 = extern_kernels.convolution(buf94, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf95, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf96 = buf95
        del buf95
        triton_poi_fused_relu_1[grid(2097152)](buf96, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf97 = extern_kernels.convolution(buf96, primals_4, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf97, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf98 = buf97
        del buf97
        triton_poi_fused_add_relu_2[grid(2097152)](buf98, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf99 = extern_kernels.convolution(buf98, primals_3, stride=(1, 1),
            padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf99, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf100 = buf99
        del buf99
        triton_poi_fused_relu_1[grid(2097152)](buf100, 2097152, XBLOCK=512,
            num_warps=8, num_stages=1)
        buf101 = extern_kernels.convolution(buf100, primals_4, stride=(1, 1
            ), padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf101, (4, 128, 64, 64), (524288, 4096, 64, 1))
        buf102 = buf101
        del buf101
        triton_poi_fused_add_relu_2[grid(2097152)](buf102, buf2, 2097152,
            XBLOCK=512, num_warps=8, num_stages=1)
        buf103 = extern_kernels.convolution(buf102, primals_5, stride=(1, 1
            ), padding=(1, 1), dilation=(1, 1), transposed=False,
            output_padding=(0, 0), groups=1, bias=None)
        assert_size_stride(buf103, (4, 1, 64, 64), (4096, 4096, 64, 1))
        buf104 = buf103
        del buf103
        triton_poi_fused_add_3[grid(16384)](buf104, buf0, primals_1, 16384,
            XBLOCK=256, num_warps=4, num_stages=1)
        del primals_1
    return (buf104, primals_2, primals_3, primals_4, primals_5, buf0, buf2,
        buf4, buf6, buf8, buf10, buf12, buf14, buf16, buf18, buf20, buf22,
        buf24, buf26, buf28, buf30, buf32, buf34, buf36, buf38, buf40,
        buf42, buf44, buf46, buf48, buf50, buf52, buf54, buf56, buf58,
        buf60, buf62, buf64, buf66, buf68, buf70, buf72, buf74, buf76,
        buf78, buf80, buf82, buf84, buf86, buf88, buf90, buf92, buf94,
        buf96, buf98, buf100, buf102)


class DRRNNew(nn.Module):

    def __init__(self):
        super(DRRNNew, self).__init__()
        self.input = nn.Conv2d(in_channels=1, out_channels=128, kernel_size
            =3, stride=1, padding=1, bias=False)
        self.conv1 = nn.Conv2d(in_channels=128, out_channels=128,
            kernel_size=3, stride=1, padding=1, bias=False)
        self.conv2 = nn.Conv2d(in_channels=128, out_channels=128,
            kernel_size=3, stride=1, padding=1, bias=False)
        self.output = nn.Conv2d(in_channels=128, out_channels=1,
            kernel_size=3, stride=1, padding=1, bias=False)
        self.relu = nn.ReLU(inplace=True)
        for m in self.modules():
            if isinstance(m, nn.Conv2d):
                n = m.kernel_size[0] * m.kernel_size[1] * m.out_channels
                m.weight.data.normal_(0, sqrt(2.0 / n))

    def forward(self, input_0):
        primals_2 = self.input.weight
        primals_3 = self.conv1.weight
        primals_4 = self.conv2.weight
        primals_5 = self.output.weight
        primals_1 = input_0
        output = call([primals_1, primals_2, primals_3, primals_4, primals_5])
        return output[0]


Measured latency_ms: 7.88932