From e55d645e2c91b18437f615e482c9d4702499bd8e Mon Sep 17 00:00:00 2001
From: Steve Malton <stephen.malton@nanoporetech.com>
Date: Wed, 6 Dec 2023 10:15:38 +0000
Subject: [PATCH 1/2] Remove unused headers

---
 dorado/nn/ModelRunner.h | 2 --
 1 file changed, 2 deletions(-)

diff --git a/dorado/nn/ModelRunner.h b/dorado/nn/ModelRunner.h
index f7d51079d..625398fc2 100644
--- a/dorado/nn/ModelRunner.h
+++ b/dorado/nn/ModelRunner.h
@@ -5,8 +5,6 @@
 #include "decode/Decoder.h"
 #include "utils/stats.h"
 
-#include <spdlog/spdlog.h>
-#include <toml.hpp>
 #include <torch/nn.h>
 
 #include <atomic>

From 525e297ad690d2b294f29c5a1c3a949109e286ec Mon Sep 17 00:00:00 2001
From: Steve Malton <stephen.malton@nanoporetech.com>
Date: Wed, 6 Dec 2023 10:17:11 +0000
Subject: [PATCH 2/2] Add warning if CPU basecalling is requested on TX2

---
 dorado/nn/Runners.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/dorado/nn/Runners.cpp b/dorado/nn/Runners.cpp
index 6e0c5ec82..c4787fe02 100644
--- a/dorado/nn/Runners.cpp
+++ b/dorado/nn/Runners.cpp
@@ -15,6 +15,7 @@
 #endif  // DORADO_GPU_BUILD
 
 #include <cxxpool.h>
+#include <spdlog/spdlog.h>
 
 #include <thread>
 
@@ -39,6 +40,10 @@ std::pair<std::vector<dorado::Runner>, size_t> create_basecall_runners(
     size_t num_devices = 1;
 
     if (device == "cpu") {
+#ifdef DORADO_TX2
+        spdlog::warn("CPU basecalling is not supported on this platform. Results may be incorrect");
+#endif  // #ifdef DORADO_TX2
+
         if (batch_size == 0) {
             batch_size = 128;
         }