Merge pull request #13 from tenstorrent/milant/adding_target_for_gene…

…rating_stubs

This PR adds target which can be used to generate python stubs.
To invoke run `cmake --build build -- make_stubs`

Removing stubs which are not used anymore.

fixes #9

CMakeLists.txt

@@ -48,3 +48,11 @@ set(SHARED_LIB_FLAGS -fPIC)
 add_subdirectory(third_party)
 add_subdirectory(pybuda)
 
+### Generate stubs for ttforge
+### Run `cmake --build build -- make_stubs` to generate stubs
+add_custom_target(make_stubs
+    COMMAND pip install mypy
+    COMMAND stubgen -m pybuda._C -m pybuda._C.autograd -m pybuda._C.graph -m pybuda._C.torch_device -m pybuda._C.runtime -o pybuda -v
+    COMMENT "Generating stubs for ttforge"
+    USES_TERMINAL
+    WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR})

pybuda/pybuda/_C/__init__.pyi

@@ -1,4 +1,4 @@
-from . import autograd as autograd, graph as graph, torch_device as torch_device
+from . import autograd as autograd, graph as graph, runtime as runtime, torch_device as torch_device
 from typing import ClassVar
 
 BLACKHOLE: Arch