Update to PyTorch 1.2 (#192)

.travis.yml

@@ -12,7 +12,7 @@ cache:
 
 install:
     - pip install -U pip
-    - pip install https://download.pytorch.org/whl/cpu/torch-1.1.0-cp36-cp36m-linux_x86_64.whl
+    - pip install torch==1.2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 
     # Keep track of Pyro dev branch
     - pip install https://github.com/pyro-ppl/pyro/archive/dev.zip

funsor/testing.py

@@ -72,7 +72,7 @@ def assert_close(actual, expected, atol=1e-6, rtol=1e-6):
     elif isinstance(actual, torch.Tensor):
         assert actual.dtype == expected.dtype, msg
         assert actual.shape == expected.shape, msg
-        if actual.dtype in (torch.long, torch.uint8):
+        if actual.dtype in (torch.long, torch.uint8, torch.bool):
             assert (actual == expected).all(), msg
         else:
             eq = (actual == expected)

funsor/torch.py

@@ -746,7 +746,7 @@ def _exp(x):
 
 @ops.log.register(torch.Tensor)
 def _log(x):
-    if x.dtype in (torch.uint8, torch.long):
+    if x.dtype in (torch.bool, torch.uint8, torch.long):
         x = x.float()
     return x.log()
 

test/test_torch.py

@@ -13,7 +13,7 @@
 
 
 @pytest.mark.parametrize('shape', [(), (4,), (3, 2)])
-@pytest.mark.parametrize('dtype', [torch.float, torch.long, torch.uint8])
+@pytest.mark.parametrize('dtype', [torch.float, torch.long, torch.uint8, torch.bool])
 def test_to_funsor(shape, dtype):
     t = torch.randn(shape).type(dtype)
     f = funsor.to_funsor(t)