You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@iotdb.apache.org by yo...@apache.org on 2026/06/02 13:49:07 UTC
(iotdb) 01/01: Remove Chronos2 pin memory option
This is an automated email from the ASF dual-hosted git repository.
CRZbulabula pushed a commit to branch remove-pin-memory
in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 80339eab035858a8db013c943f407b14a50fe1c0
Author: Yongzao <53...@qq.com>
AuthorDate: Tue Jun 2 21:48:26 2026 +0800
Remove Chronos2 pin memory option
---
.../iotdb/ainode/it/AINodeSharedClusterIT.java | 22 ++++++++++++++++++++++
.../core/model/chronos2/pipeline_chronos2.py | 1 -
2 files changed, 22 insertions(+), 1 deletion(-)
diff --git a/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeSharedClusterIT.java b/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeSharedClusterIT.java
index 4ea2b4af41a..27dab22a580 100644
--- a/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeSharedClusterIT.java
+++ b/integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeSharedClusterIT.java
@@ -50,6 +50,7 @@ import java.util.List;
import java.util.Set;
import java.util.concurrent.TimeUnit;
+import static org.apache.iotdb.ainode.utils.AINodeTestUtils.BUILTIN_LTSM_MAP;
import static org.apache.iotdb.ainode.utils.AINodeTestUtils.BUILTIN_MODEL_MAP;
import static org.apache.iotdb.ainode.utils.AINodeTestUtils.checkHeader;
import static org.apache.iotdb.ainode.utils.AINodeTestUtils.checkModelNotOnSpecifiedDevice;
@@ -438,6 +439,27 @@ public class AINodeSharedClusterIT {
// ========== Concurrent forecast tests ==========
+ @Test
+ public void largeTimeSeriesModelLoadDevicePlacementTest()
+ throws SQLException, InterruptedException {
+ try (Connection connection = EnvFactory.getEnv().getConnection(BaseEnv.TABLE_SQL_DIALECT);
+ Statement statement = connection.createStatement()) {
+ for (FakeModelInfo modelInfo : BUILTIN_LTSM_MAP.values()) {
+ loadAndUnloadModelOnDevices(statement, modelInfo.getModelId(), "cpu");
+ loadAndUnloadModelOnDevices(statement, modelInfo.getModelId(), "0");
+ loadAndUnloadModelOnDevices(statement, modelInfo.getModelId(), "cpu,0");
+ }
+ }
+ }
+
+ private void loadAndUnloadModelOnDevices(Statement statement, String modelId, String devices)
+ throws SQLException, InterruptedException {
+ statement.execute(String.format("LOAD MODEL %s TO DEVICES '%s'", modelId, devices));
+ checkModelOnSpecifiedDevice(statement, modelId, devices);
+ statement.execute(String.format("UNLOAD MODEL %s FROM DEVICES '%s'", modelId, devices));
+ checkModelNotOnSpecifiedDevice(statement, modelId, devices);
+ }
+
@Test
public void concurrentForecastTest() throws SQLException, InterruptedException {
for (FakeModelInfo modelInfo : CONCURRENT_FORECAST_MODELS) {
diff --git a/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py b/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py
index 01ff78ba48d..083125e4b99 100644
--- a/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py
+++ b/iotdb-core/ainode/iotdb/ainode/core/model/chronos2/pipeline_chronos2.py
@@ -309,7 +309,6 @@ class Chronos2Pipeline(ForecastPipeline):
test_loader = DataLoader(
test_dataset,
batch_size=None,
- pin_memory=True,
shuffle=False,
drop_last=False,
)