From 365b612ec5bfa5ced609edb236dde9e51875f861 Mon Sep 17 00:00:00 2001 From: Sterling Date: Mon, 26 Aug 2024 10:48:07 -0500 Subject: [PATCH] Adding SPDX headers to missing files --- pybuda/test/mlir/llama/utils/utils.py | 4 ++++ pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py | 4 ++++ pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py | 4 ++++ 3 files changed, 12 insertions(+) diff --git a/pybuda/test/mlir/llama/utils/utils.py b/pybuda/test/mlir/llama/utils/utils.py index 03751ccb0..a3b7f1d78 100644 --- a/pybuda/test/mlir/llama/utils/utils.py +++ b/pybuda/test/mlir/llama/utils/utils.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC + +# SPDX-License-Identifier: Apache-2.0 + from transformers import LlamaConfig, LlamaForCausalLM, LlamaTokenizer import pybuda diff --git a/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py b/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py index eb0364aae..dba714365 100644 --- a/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py +++ b/pybuda/test/mlir/mnist/training/mnist_linear_pybuda.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC + +# SPDX-License-Identifier: Apache-2.0 + import torch from torchvision import datasets, transforms from torch.utils.tensorboard import SummaryWriter diff --git a/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py b/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py index 7958745e4..a9a4b8f92 100644 --- a/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py +++ b/pybuda/test/mlir/mnist/training/mnist_linear_pytorch.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: © 2024 Tenstorrent AI ULC + +# SPDX-License-Identifier: Apache-2.0 + import torch from torchvision import datasets, transforms from torch.utils.tensorboard import SummaryWriter