PyTorch 1.7 compatibility

Summary: Small changes discovered based on circleCI failures.

Reviewed By: patricklabatut

Differential Revision: D34426807

fbshipit-source-id: 819860f34b2f367dd24057ca7490284204180a13
This commit is contained in:
Jeremy Reizenstein
2022-02-25 07:53:34 -08:00
committed by Facebook GitHub Bot
parent f816568735
commit 4d043fc9ac
9 changed files with 34 additions and 23 deletions

View File

@@ -10,6 +10,7 @@ import unittest
import numpy as np
import torch
from common_testing import TestCaseMixin
from pytorch3d.common.compat import meshgrid_ij
from pytorch3d.ops import eyes
from pytorch3d.renderer import (
AlphaCompositor,
@@ -129,8 +130,8 @@ class TestTensorProperties(TestCaseMixin, unittest.TestCase):
point_radius = 0.015
n_pts = n_grid_pts * n_grid_pts
pts = torch.stack(
torch.meshgrid(
[torch.linspace(-grid_scale, grid_scale, n_grid_pts)] * 2, indexing="ij"
meshgrid_ij(
[torch.linspace(-grid_scale, grid_scale, n_grid_pts)] * 2,
),
dim=-1,
)