0234315776751221ef9da5d2073e32f9f54c01bb,test/color/test_gray.py,TestRgbToGrayscale,test_jit,#TestRgbToGrayscale#Any#Any#,60

Before Change

        assert gradcheck(kornia.rgb_to_grayscale, (img,), raise_exception=True)

    def test_jit(self, device, dtype):
        if "cuda" in str(device):
            pytest.skip("skip this test for a while")
        batch_size, channels, height, width = 2, 3, 64, 64
        img = torch.ones(batch_size, channels, height, width).to(device, dtype)
        op = kornia.rgb_to_grayscale
        op_jit = kornia.jit.rgb_to_grayscale

After Change

    @pytest.mark.jit
    def test_jit(self, device, dtype):
        B, C, H, W = 2, 3, 4, 4
        img = torch.ones(B, C, H, W, device=device, dtype=dtype)
        op = kornia.rgb_to_grayscale
        op_jit = kornia.jit.rgb_to_grayscale
        assert_allclose(op(img), op_jit(img))

Instances