k2-fsa · pkufool · Oct 27, 2021 · Oct 11, 2021 · Oct 12, 2021 · Oct 12, 2021
diff --git a/k2/CMakeLists.txt b/k2/CMakeLists.txt
@@ -1,2 +1,7 @@
 add_subdirectory(csrc)
-add_subdirectory(python)
+if(K2_USE_PYTORCH)
+  add_subdirectory(torch)
+else()
+  message(FATAL_ERROR "Please select a framework.")
+endif()
+# add_subdirectory(python)
-# add_subdirectory(python)
-# add_subdirectory(python)
diff --git a/k2/python/csrc/k2.cu b/k2/python/csrc/k2.cu
diff --git a/k2/python/csrc/torch.cu b/k2/python/csrc/torch.cu
diff --git a/k2/python/csrc/torch/v2/README.md b/k2/python/csrc/torch/v2/README.md
diff --git a/k2/python/csrc/torch/v2/autograd/arc_sort.h b/k2/python/csrc/torch/v2/autograd/arc_sort.h
diff --git a/k2/python/csrc/torch/v2/autograd/ragged_arc_holder.h b/k2/python/csrc/torch/v2/autograd/ragged_arc_holder.h
diff --git a/k2/python/csrc/torch/v2/fsa.cu b/k2/python/csrc/torch/v2/fsa.cu
diff --git a/k2/python/csrc/torch/v2/k2.cu b/k2/python/csrc/torch/v2/k2.cu