remove keops and taichi

kentechx · Jul 24, 2023 · 313a8d2 · 313a8d2
1 parent 26256f8
commit 313a8d2
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 16 deletions.
diff --git a/pointnext/pointnext.py b/pointnext/pointnext.py
@@ -1,6 +1,5 @@
 from collections import namedtuple
 from typing import Union
-from pykeops.torch import LazyTensor
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -9,20 +8,6 @@
 # from .utils import farthest_point_sampling, ball_query_pytorch
 from .ops import ball_query, furthest_point_sample, three_interpolation
 
-__TAICHI__ = False
-__KEOPS__ = True
-
-
-def enable_taichi():
-    import taichi as ti
-    global __TAICHI__
-    __TAICHI__ = True
-    ti.init(ti.cuda)
-
-
-def disable_keops():
-    global __KEOPS__
-    __KEOPS__ = False
 
 
 def exists(val):

diff --git a/setup.py b/setup.py
@@ -40,7 +40,6 @@
     install_requires=[
         'torch>=1.10',
         'einops>=0.6.1',
-        'pykeops>=2.1.2',
     ],
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension},