Skip to content

Commit

Permalink
#4764: Add support for FD targetting ethernet cores
Browse files Browse the repository at this point in the history
- change app fw to start and finish on device init and close
  • Loading branch information
aliuTT committed Jan 27, 2024
1 parent 26cf491 commit 9a3d46c
Show file tree
Hide file tree
Showing 17 changed files with 1,740 additions and 202 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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_;
};
Loading

0 comments on commit 9a3d46c

Please sign in to comment.