Skip to content

Commit

Permalink
add copyrights
Browse files Browse the repository at this point in the history
  • Loading branch information
siddharth9820 committed Jul 9, 2024
1 parent 0d7b577 commit 4b0d1ac
Show file tree
Hide file tree
Showing 6 changed files with 30 additions and 0 deletions.
5 changes: 5 additions & 0 deletions axonn/intra_layer/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
# See the top-level LICENSE file for details.
#
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

from contextlib import contextmanager
from .fully_connected import Linear # noqa: F401
from .conv import Conv2d # noqa: F401
Expand Down
5 changes: 5 additions & 0 deletions axonn/intra_layer/embedding.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
# See the top-level LICENSE file for details.
#
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import torch.distributed as dist
import torch
import torch.nn.functional as F
Expand Down
5 changes: 5 additions & 0 deletions axonn/intra_layer/gradient_normalization.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
# See the top-level LICENSE file for details.
#
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import torch

# for backwards compatibility with pytorch 1.13
Expand Down
5 changes: 5 additions & 0 deletions axonn/tests/test_intra_layer_conv.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
# See the top-level LICENSE file for details.
#
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import torch
import pytest
from mpi4py import MPI # noqa: F401
Expand Down
5 changes: 5 additions & 0 deletions axonn/tests/test_intra_layer_emb.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
# See the top-level LICENSE file for details.
#
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import torch
import pytest
from axonn import axonn as ax
Expand Down
5 changes: 5 additions & 0 deletions axonn/tests/test_intra_layer_fc.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,8 @@
# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
# See the top-level LICENSE file for details.
#
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

import torch
import pytest
from axonn import axonn as ax
Expand Down

0 comments on commit 4b0d1ac

Please # to comment.