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 authored and TT-BrianLiu committed Jan 26, 2024
1 parent 3dfbec6 commit cb2e63c
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 cb2e63c

Please sign in to comment.