abreza commited on
Commit
d86d3e4
·
1 Parent(s): 2421544
Files changed (1) hide show
  1. app.py +24 -22
app.py CHANGED
@@ -12,28 +12,30 @@ import csv
12
  import spaces
13
  # from triton.fb import build_paths
14
 
15
- from torch.utils.cpp_extension import load
16
-
17
- # Compile and load the CUDA extension
18
- adaptive_conv_cuda_impl = load(
19
- name='adaptive_conv_cuda_impl',
20
- sources=[
21
- 'featup/adaptive_conv_cuda/adaptive_conv_cuda.cpp',
22
- 'featup/adaptive_conv_cuda/adaptive_conv_kernel.cu'
23
- ],
24
- extra_cflags=['-O3'],
25
- extra_cuda_cflags=['-O3']
26
- )
27
-
28
- # Compile and load the C++ extension
29
- adaptive_conv_cpp_impl = load(
30
- name='adaptive_conv_cpp_impl',
31
- sources=['featup/adaptive_conv_cuda/adaptive_conv.cpp'],
32
- extra_cflags=['-O3']
33
- )
34
-
35
- print(adaptive_conv_cuda_impl, adaptive_conv_cpp_impl)
36
-
 
 
37
 
38
  def plot_feats(image, lr, hr):
39
  from featup.util import pca, remove_axes
 
12
  import spaces
13
  # from triton.fb import build_paths
14
 
15
+ # from torch.utils.cpp_extension import load
16
+
17
+ # # Compile and load the CUDA extension
18
+ # adaptive_conv_cuda_impl = load(
19
+ # name='adaptive_conv_cuda_impl',
20
+ # sources=[
21
+ # 'featup/adaptive_conv_cuda/adaptive_conv_cuda.cpp',
22
+ # 'featup/adaptive_conv_cuda/adaptive_conv_kernel.cu'
23
+ # ],
24
+ # extra_cflags=['-O3'],
25
+ # extra_cuda_cflags=['-O3']
26
+ # )
27
+
28
+ # # Compile and load the C++ extension
29
+ # adaptive_conv_cpp_impl = load(
30
+ # name='adaptive_conv_cpp_impl',
31
+ # sources=['featup/adaptive_conv_cuda/adaptive_conv.cpp'],
32
+ # extra_cflags=['-O3']
33
+ # )
34
+
35
+ # print(adaptive_conv_cuda_impl, adaptive_conv_cpp_impl)
36
+
37
+ output = subprocess.check_output(['ls', '~/local/']).decode()
38
+ print(output)
39
 
40
  def plot_feats(image, lr, hr):
41
  from featup.util import pca, remove_axes