#4764: Add support for FD targetting ethernet cores

- change app fw to start and finish on device init and close
tenstorrent · Jan 25, 2024 · c04e77d · c04e77d
1 parent 8c8056d
commit c04e77d
Show file tree

Hide file tree

Showing 17 changed files with 1,748 additions and 210 deletions.
diff --git a/tests/tt_metal/tt_metal/unit_tests_fast_dispatch/common/command_queue_fixture.hpp b/tests/tt_metal/tt_metal/unit_tests_fast_dispatch/common/command_queue_fixture.hpp
@@ -63,3 +63,35 @@ class CommandQueueMultiDeviceFixture : public ::testing::Test {
     tt::ARCH arch_;
     size_t num_devices_;
 };
+
+class CommandQueuePCIDevicesFixture : public ::testing::Test {
+   protected:
+    void SetUp() override {
+        auto slow_dispatch = getenv("TT_METAL_SLOW_DISPATCH_MODE");
+        if (slow_dispatch) {
+            TT_THROW("This suite can only be run with fast dispatch or TT_METAL_SLOW_DISPATCH_MODE unset");
+            GTEST_SKIP();
+        }
+        arch_ = tt::get_arch_from_string(tt::test_utils::get_env_arch_name());
+
+        num_devices_ = tt::tt_metal::GetNumPCIeDevices();
+        if (num_devices_ < 2) {
+            GTEST_SKIP();
+        }
+
+        for (unsigned int id = 0; id < num_devices_; id++) {
+            auto* device = tt::tt_metal::CreateDevice(id);
+            devices_.push_back(device);
+        }
+    }
+
+    void TearDown() override {
+        for (unsigned int id = 0; id < devices_.size(); id++) {
+            tt::tt_metal::CloseDevice(devices_.at(id));
+        }
+    }
+
+    std::vector<tt::tt_metal::Device*> devices_;
+    tt::ARCH arch_;
+    size_t num_devices_;
+};