Add an ad-hoc interface to Yateto

Author: davschneller

tensorforge/__init__.py

@@ -1,4 +1,6 @@
 from .type import *
 from .generator import NamespacedGenerator, Generator, simpleParameterSpace, parameterSpaceFromRanges
 from .arch import useArchitectureIdentifiedBy
-from .gemm_configuration import *
+from .gemm_configuration import *
+
+from .interop import *

tensorforge/codegen/gpukernel.py

@@ -211,6 +211,16 @@ def __call__(self, routineName, fileName):
 
     return self._generator.get_header()
 
+class GpuKernelRoutineGenerator:
+  def __init__(self, arch):
+    self.generator = GpuKernelGenerator(arch)
+
+  def generate(self, cpp, cache):
+    self.generator.generate(cpp, cache)
+
+  def add_operation(self, dest, ops, target, permute, add):
+    self.generator.add_operation(dest, ops, target, permute, add)
+
 
 class GpuKernelFactory(KernelFactory):
   def __init__(self, cpp, arch, target):

tensorforge/interop.py

@@ -0,0 +1,20 @@
+def get_cmake_path():
+    import os
+    mydir = os.path.dirname(os.path.realpath(__file__))
+    return os.path.join(mydir, 'share', 'cmake')
+
+def print_cmake_path():
+    print(get_cmake_path())
+
+def get_routine_generator(yateto):
+    import tensorforge.codegen.gpukernel as gk
+    return gk.GpuKernelRoutineGenerator
+
+def get_version():
+    import os
+    mydir = os.path.dirname(os.path.realpath(__file__))
+    with file(os.path.join(mydir, 'VERSION')):
+        return file.read()
+
+def print_version():
+    print(get_version())

tensorforge/share/cmake/TensorForgeConfigVersion.cmake

@@ -1,4 +1,4 @@
-execute_process(COMMAND python3 -c "import tensorforge.support; tensorforge.support.get_version()"
+execute_process(COMMAND python3 -c "import tensorforge; tensorforge.get_version()"
                 OUTPUT_VARIABLE PACKAGE_VERSION)
 
 # Check whether the requested PACKAGE_FIND_VERSION is compatible