def METHOD_NAME(self):
    @mmcv.jit
    def func(a, b):
        return torch.mean((a - b) * (a - b))
    assert len(func._cache._cache) == 0
    x = torch.rand((16, 32))
    y = torch.rand((16, 32))
    func(x, y)
    assert len(func._cache._cache) == 1
    func(x, b=y)
    assert len(func._cache._cache) == 1
    func(b=y, a=x)
    assert len(func._cache._cache) == 1