Skip to content

Commit 4cf5430

Browse files
committed
Update AINodeConcurrentForecastIT.java
1 parent 920df32 commit 4cf5430

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

integration-test/src/test/java/org/apache/iotdb/ainode/it/AINodeConcurrentForecastIT.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,10 @@ private static void prepareDataForTableModel() throws SQLException {
7979
@Test
8080
public void concurrentGPUForecastTest() throws SQLException, InterruptedException {
8181
for (AINodeTestUtils.FakeModelInfo modelInfo : BUILTIN_LTSM_MAP.values()) {
82+
if (modelInfo.getModelId().equals("chronos2")) {
83+
// TODO: @yunbow30944 Support chronos2 model concurrent inference in GPU
84+
continue;
85+
}
8286
concurrentGPUForecastTest(modelInfo);
8387
}
8488
}

0 commit comments

Comments
 (0)