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:06 UTC
(iotdb) branch remove-pin-memory created (now 80339eab035)
This is an automated email from the ASF dual-hosted git repository.
CRZbulabula pushed a change to branch remove-pin-memory
in repository https://gitbox.apache.org/repos/asf/iotdb.git
at 80339eab035 Remove Chronos2 pin memory option
This branch includes the following new commits:
new 80339eab035 Remove Chronos2 pin memory option
The 1 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
(iotdb) 01/01: Remove Chronos2 pin memory option
Posted by yo...@apache.org.
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,
)