Removed the reference to Metalium::Pybinds in tt-train

tenstorrent · Feb 25, 2025 · cc4ad61 · cc4ad61
1 parent 4c30c6f
commit cc4ad61
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tt-train/sources/ttml/CMakeLists.txt b/tt-train/sources/ttml/CMakeLists.txt
@@ -66,9 +66,9 @@ if(NOT TARGET Metalium::Metal)
         message(FATAL_ERROR "libtt_metal.so not found in ${METALIUM_LIB_PATH}")
     endif()
     if(TTNN_LIBRARY)
-        add_library(Metalium::PYBIND SHARED IMPORTED)
+        add_library(Metalium::TTNN SHARED IMPORTED)
         set_target_properties(
-            Metalium::PYBIND
+            Metalium::TTNN
             PROPERTIES
                 IMPORTED_LOCATION
                     "${TTNN_LIBRARY}"
@@ -101,7 +101,7 @@ target_link_libraries(
         pthread
         atomic
         Metalium::Metal
-        Metalium::PYBIND
+        Metalium::TTNN
         Python::Python
         fmt::fmt-header-only
         magic_enum::magic_enum