#4003: restructure ttnn

tenstorrent · Jan 25, 2024 · 9ecb68f · 9ecb68f
1 parent e1c9731
commit 9ecb68f
Show file tree

Hide file tree

Showing 45 changed files with 1,553 additions and 1,502 deletions.
diff --git a/docs/aspell-dictionary.pws b/docs/aspell-dictionary.pws
@@ -1,4 +1,4 @@
-personal_ws-1.1 en 455
+personal_ws-1.1 en 457
 ABI
 ADDI
 API
@@ -137,6 +137,7 @@ TRISC
 TSLICE
 TTNN
 TTNN's
+TensTorrent's
 TensorMemoryLayout
 Tensorflow
 Tenstorrent
@@ -421,6 +422,7 @@ triscs
 tt
 ttDNN
 ttnn
+ttnn's
 txt
 ublock
 ublocks

diff --git a/docs/source/ttnn/about.rst b/docs/source/ttnn/about.rst
@@ -1,9 +1,10 @@
-What is TTNN?
+What is ttnn?
 #############
 
-TTNN is a Python library that provides a user-friendly interface to ``TT-METAL``.
-The TTNN library assumes the user is familiar with `PyTorch <https://pytorch.org/>`_ and provides
-operations that easily translate `PyTorch <https://pytorch.org/>`_ tensors to ``ttnn.Tensor(s)``.
-TTNN's primary dependency is :doc:`TT-LIB </ttnn/dependencies/tt_lib>`.
+ttnn is a library that provides a user-friendly interface to operations that run on TensTorrent's hardware using ``tt-metal`` programming model.
+
+ttnn is designed be intuitive to an user that is familiar with `PyTorch <https://pytorch.org/>`_.
+
+ttnn's primary dependency is :doc:`TT-LIB </ttnn/dependencies/tt_lib>` which provides the implementation for all of the operations used by ttnn.
 
 We trust that this library will be valuable to helping you on your journey to take full advantage of our devices!
diff --git a/module.mk b/module.mk
@@ -137,13 +137,15 @@ LIBS_TO_BUILD += \
 	tools \
 	tt_metal \
 	tracy \
-	tt_eager
+	tt_eager \
+	ttnn/build
 
 # These must be in dependency order (enforces no circular deps)
 include $(UMD_HOME)/device/module.mk
 include $(TT_METAL_HOME)/tt_metal/common/common.mk
 include $(TT_METAL_HOME)/tt_metal/module.mk
 include $(TT_METAL_HOME)/tt_eager/module.mk
+include $(TT_METAL_HOME)/ttnn/module.mk
 include $(TT_METAL_HOME)/tt_metal/python_env/module.mk
 include $(TT_METAL_HOME)/tests/module.mk
 

diff --git a/setup.py b/setup.py
@@ -155,7 +155,7 @@ def is_editable_install_(self):
 # Include tt_metal_C for kernels and src/ and tools
 # And any kernels inside `tt_eager/tt_dnn. We must keep all ops kernels inside
 # tt_dnn
-packages = ["tt_lib", "tt_lib.tt_metal", "tt_lib.models", "tt_lib.scripts", "tt_lib.tt_eager.tt_dnn", "ttnn"]
+packages = ["tt_lib", "tt_lib.tt_metal", "tt_lib.models", "tt_lib.scripts", "tt_lib.tt_eager.tt_dnn"]
 
 # Empty sources in order to force a BUDAEagerBuild execution
 buda_eager_lib_C = Extension("tt_lib._C", sources=[])
@@ -172,7 +172,6 @@ def is_editable_install_(self):
         "tt_lib.models": "models",
         "tt_lib.scripts": "scripts",
         "tt_lib.tt_eager.tt_dnn": "tt_eager/tt_dnn",
-        "ttnn": "ttnn",
     },
     include_package_data=True,
     long_description_content_type="text/markdown",

diff --git a/tests/ttnn/unit_tests/test_add.py → tests/ttnn/unit_tests/operations/test_add.py b/tests/ttnn/unit_tests/test_add.py → tests/ttnn/unit_tests/operations/test_add.py
diff --git a/tests/ttnn/unit_tests/test_concat.py → ...ttnn/unit_tests/operations/test_concat.py b/tests/ttnn/unit_tests/test_concat.py → ...ttnn/unit_tests/operations/test_concat.py
diff --git a/tests/ttnn/unit_tests/test_conv.py → ...s/ttnn/unit_tests/operations/test_conv.py b/tests/ttnn/unit_tests/test_conv.py → ...s/ttnn/unit_tests/operations/test_conv.py
diff --git a/tests/ttnn/unit_tests/test_embedding.py → ...n/unit_tests/operations/test_embedding.py b/tests/ttnn/unit_tests/test_embedding.py → ...n/unit_tests/operations/test_embedding.py
diff --git a/tests/ttnn/unit_tests/test_group_norm.py → .../unit_tests/operations/test_group_norm.py b/tests/ttnn/unit_tests/test_group_norm.py → .../unit_tests/operations/test_group_norm.py
diff --git a/tests/ttnn/unit_tests/test_layer_norm.py → .../unit_tests/operations/test_layer_norm.py b/tests/ttnn/unit_tests/test_layer_norm.py → .../unit_tests/operations/test_layer_norm.py
diff --git a/tests/ttnn/unit_tests/test_matmul.py → ...ttnn/unit_tests/operations/test_matmul.py b/tests/ttnn/unit_tests/test_matmul.py → ...ttnn/unit_tests/operations/test_matmul.py
diff --git a/tests/ttnn/unit_tests/test_max_pool.py → ...nn/unit_tests/operations/test_max_pool.py b/tests/ttnn/unit_tests/test_max_pool.py → ...nn/unit_tests/operations/test_max_pool.py
diff --git a/tests/ttnn/unit_tests/test_mean.py → ...s/ttnn/unit_tests/operations/test_mean.py b/tests/ttnn/unit_tests/test_mean.py → ...s/ttnn/unit_tests/operations/test_mean.py
diff --git a/tests/ttnn/unit_tests/test_mul.py → tests/ttnn/unit_tests/operations/test_mul.py b/tests/ttnn/unit_tests/test_mul.py → tests/ttnn/unit_tests/operations/test_mul.py
diff --git a/tests/ttnn/unit_tests/test_pad.py → tests/ttnn/unit_tests/operations/test_pad.py b/tests/ttnn/unit_tests/test_pad.py → tests/ttnn/unit_tests/operations/test_pad.py
diff --git a/tests/ttnn/unit_tests/test_pad_to_tile.py → ...unit_tests/operations/test_pad_to_tile.py b/tests/ttnn/unit_tests/test_pad_to_tile.py → ...unit_tests/operations/test_pad_to_tile.py
diff --git a/tests/ttnn/unit_tests/test_permute.py → ...tnn/unit_tests/operations/test_permute.py b/tests/ttnn/unit_tests/test_permute.py → ...tnn/unit_tests/operations/test_permute.py
diff --git a/...ttnn/unit_tests/test_repeat_interleave.py → ...ests/operations/test_repeat_interleave.py b/...ttnn/unit_tests/test_repeat_interleave.py → ...ests/operations/test_repeat_interleave.py
diff --git a/tests/ttnn/unit_tests/test_rms_norm.py → ...nn/unit_tests/operations/test_rms_norm.py b/tests/ttnn/unit_tests/test_rms_norm.py → ...nn/unit_tests/operations/test_rms_norm.py
diff --git a/tests/ttnn/unit_tests/test_softmax.py → ...tnn/unit_tests/operations/test_softmax.py b/tests/ttnn/unit_tests/test_softmax.py → ...tnn/unit_tests/operations/test_softmax.py
diff --git a/tests/ttnn/unit_tests/test_split.py → .../ttnn/unit_tests/operations/test_split.py b/tests/ttnn/unit_tests/test_split.py → .../ttnn/unit_tests/operations/test_split.py
diff --git a/tests/ttnn/unit_tests/test_sub.py → tests/ttnn/unit_tests/operations/test_sub.py b/tests/ttnn/unit_tests/test_sub.py → tests/ttnn/unit_tests/operations/test_sub.py
diff --git a/tests/ttnn/unit_tests/test_to_layout.py → ...n/unit_tests/operations/test_to_layout.py b/tests/ttnn/unit_tests/test_to_layout.py → ...n/unit_tests/operations/test_to_layout.py
diff --git a/tests/ttnn/unit_tests/test_transformer.py → ...unit_tests/operations/test_transformer.py b/tests/ttnn/unit_tests/test_transformer.py → ...unit_tests/operations/test_transformer.py
diff --git a/tests/ttnn/unit_tests/test_unary.py → .../ttnn/unit_tests/operations/test_unary.py b/tests/ttnn/unit_tests/test_unary.py → .../ttnn/unit_tests/operations/test_unary.py
diff --git a/...s/ttnn/unit_tests/test_unpad_from_tile.py → ..._tests/operations/test_unpad_from_tile.py b/...s/ttnn/unit_tests/test_unpad_from_tile.py → ..._tests/operations/test_unpad_from_tile.py
diff --git a/ttnn/binary.py b/ttnn/binary.py