diff --git a/tests/pytorch/include/img.hpp b/tests/pytorch/include/img.hpp
index 9a2163308..f839df342 100644
--- a/tests/pytorch/include/img.hpp
+++ b/tests/pytorch/include/img.hpp
@@ -15,7 +15,9 @@
 //===----------------------------------------------------------------------===//
 //
 // original source from https://github.com/buddy-compiler/buddy-mlir
+// Modified for embedded environment.
 //
+//===----------------------------------------------------------------------===//
 
 #ifndef IMG_H
 #define IMG_H
diff --git a/tests/pytorch/include/memref.hpp b/tests/pytorch/include/memref.hpp
index 4875a383f..95e6725f2 100644
--- a/tests/pytorch/include/memref.hpp
+++ b/tests/pytorch/include/memref.hpp
@@ -1,3 +1,24 @@
+//===- ImgContainer.h -----------------------------------------------------===//
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+//===----------------------------------------------------------------------===//
+//
+// Original source from https://github.com/buddy-compiler/buddy-mlir.
+// Modified for embedded environment.
+//
+//===----------------------------------------------------------------------===//
+
 #ifndef MEMREF_H
 #define MEMREF_H
 
diff --git a/tests/pytorch/lib/MemrefCopy.cc b/tests/pytorch/lib/MemrefCopy.cc
index 0fef5693b..c7f15ff79 100644
--- a/tests/pytorch/lib/MemrefCopy.cc
+++ b/tests/pytorch/lib/MemrefCopy.cc
@@ -1,3 +1,15 @@
+//===- CRunnerUtils.cpp - Utils for MLIR execution ------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+// Modified to basic data type for linkage usage only.
+//
+//===----------------------------------------------------------------------===//
+
 #include <stdint.h>
 #include <stdlib.h>
 #include <string.h>