From f2b229c1d1cfae0dc61ba4390bac626ba1281960 Mon Sep 17 00:00:00 2001
From: Georgia Gkioxari <gkioxari@fb.com>
Date: Mon, 20 Apr 2020 23:02:25 -0700
Subject: [PATCH] pytorch3d compatibility

Summary: Making meshrcnn compatible with new PyTorch3D features/API changes.

Reviewed By: nikhilaravi

Differential Revision: D21149516

fbshipit-source-id: 1c7b8c1c1f5a2abe7d379fee10ded5d2db21515a
---
 pytorch3d/ops/knn.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pytorch3d/ops/knn.py b/pytorch3d/ops/knn.py
index 3ec6e3f0..aff9fc6f 100644
--- a/pytorch3d/ops/knn.py
+++ b/pytorch3d/ops/knn.py
@@ -145,6 +145,9 @@ def knn_points(
     if p1.shape[2] != p2.shape[2]:
         raise ValueError("pts1 and pts2 must have the same point dimension.")
 
+    p1 = p1.contiguous()
+    p2 = p2.contiguous()
+
     P1 = p1.shape[1]
     P2 = p2.shape[1]