From e9a956301c96ac2b0dee07f1441af43d9e5b4529 Mon Sep 17 00:00:00 2001
From: huangyunlong <huangyunlong4@huawei.com>
Date: Thu, 17 Oct 2024 15:20:23 +0800
Subject: [PATCH] test

---
 examples/libtorch_resnet/libtorch_resnet.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/libtorch_resnet/libtorch_resnet.cpp b/examples/libtorch_resnet/libtorch_resnet.cpp
index f6bf1675c4..06ae22e341 100644
--- a/examples/libtorch_resnet/libtorch_resnet.cpp
+++ b/examples/libtorch_resnet/libtorch_resnet.cpp
@@ -6,6 +6,7 @@
 #include <torch/script.h>
 
 #include <torch_npu/torch_npu.h>
+#include "torch_npu/csrc/core/npu/NPUStream.h"
 
 int main(int argc, const char* argv[]) {
   if (argc != 2) {
-- 
Gitee