From 3b342ff652352b6982970de89d44759cb94d0d6e Mon Sep 17 00:00:00 2001 From: Eitan Turok Date: Thu, 8 Aug 2024 20:05:47 +0000 Subject: [PATCH] fix license --- .pre-commit/FILE_HEADER | 2 +- megablocks/__init__.py | 2 +- megablocks/_version.py | 2 +- megablocks/backend/__init__.py | 2 +- megablocks/backend/kernels.py | 2 +- megablocks/benchmark_util.py | 2 +- megablocks/grouped_gemm_util.py | 2 +- megablocks/layers/__init__.py | 2 +- megablocks/layers/activation_fn.py | 2 +- megablocks/layers/all_to_all.py | 2 +- megablocks/layers/arguments.py | 2 +- megablocks/layers/common.py | 2 +- megablocks/layers/dmlp_registry.py | 2 +- megablocks/layers/dmoe.py | 2 +- megablocks/layers/gelu.py | 2 +- megablocks/layers/glu.py | 2 +- megablocks/layers/memory_test.py | 2 +- megablocks/layers/mlp.py | 2 +- megablocks/layers/moe.py | 2 +- megablocks/layers/mpu.py | 2 +- megablocks/layers/router.py | 2 +- megablocks/layers/sharedexpert_registry.py | 2 +- megablocks/layers/testing.py | 2 +- megablocks/layers/weight_parallel.py | 2 +- megablocks/ops/__init__.py | 2 +- megablocks/ops/all_to_all_benchmark.py | 2 +- megablocks/ops/binned_gather.py | 2 +- megablocks/ops/binned_scatter.py | 2 +- megablocks/ops/cumsum.py | 2 +- megablocks/ops/gather.py | 2 +- megablocks/ops/histogram.py | 2 +- megablocks/ops/histogram_benchmark.py | 2 +- megablocks/ops/matmul_benchmark.py | 2 +- megablocks/ops/padded_gather.py | 2 +- megablocks/ops/padded_scatter.py | 2 +- megablocks/ops/padded_scatter_benchmark.py | 2 +- megablocks/ops/permute_benchmark.py | 2 +- megablocks/ops/repeat.py | 2 +- megablocks/ops/replicate.py | 2 +- megablocks/ops/round_up.py | 2 +- megablocks/ops/scatter.py | 2 +- megablocks/ops/sort.py | 2 +- megablocks/ops/sort_benchmark.py | 2 +- megablocks/ops/sum.py | 2 +- megablocks/ops/topology.py | 2 +- setup.py | 2 +- tests/conftest.py | 2 +- tests/fixtures/autouse.py | 2 +- tests/fixtures/fixtures.py | 2 +- tests/layers/dmoe_test.py | 2 +- tests/layers/glu_test.py | 2 +- tests/layers/moe_test.py | 2 +- tests/layers/parallelism_test.py | 2 +- tests/ops/binned_gather_test.py | 2 +- tests/ops/binned_scatter_test.py | 2 +- tests/ops/cumsum_test.py | 2 +- tests/ops/histogram_test.py | 2 +- tests/ops/padded_gather_test.py | 2 +- tests/ops/padded_scatter_test.py | 2 +- tests/ops/replicate_test.py | 2 +- tests/ops/sort_test.py | 2 +- tests/ops/topology_test.py | 2 +- 62 files changed, 62 insertions(+), 62 deletions(-) diff --git a/.pre-commit/FILE_HEADER b/.pre-commit/FILE_HEADER index a2432cc..5081c93 100644 --- a/.pre-commit/FILE_HEADER +++ b/.pre-commit/FILE_HEADER @@ -1,2 +1,2 @@ -Copyright 2024 Databricks MegaBlocks authors +Copyright 2024 Databricks SPDX-License-Identifier: Apache-2.0 diff --git a/megablocks/__init__.py b/megablocks/__init__.py index 7f022e9..d8d1848 100644 --- a/megablocks/__init__.py +++ b/megablocks/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from megablocks.layers.arguments import Arguments diff --git a/megablocks/_version.py b/megablocks/_version.py index ffdc54a..44ea780 100644 --- a/megablocks/_version.py +++ b/megablocks/_version.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 """The MegaBlocks Version.""" diff --git a/megablocks/backend/__init__.py b/megablocks/backend/__init__.py index 1018343..9d4e43e 100644 --- a/megablocks/backend/__init__.py +++ b/megablocks/backend/__init__.py @@ -1,2 +1,2 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 diff --git a/megablocks/backend/kernels.py b/megablocks/backend/kernels.py index a25e1a4..b831826 100644 --- a/megablocks/backend/kernels.py +++ b/megablocks/backend/kernels.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/benchmark_util.py b/megablocks/benchmark_util.py index ea6069b..02612d9 100644 --- a/megablocks/benchmark_util.py +++ b/megablocks/benchmark_util.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/megablocks/grouped_gemm_util.py b/megablocks/grouped_gemm_util.py index 70b410f..07dbc04 100644 --- a/megablocks/grouped_gemm_util.py +++ b/megablocks/grouped_gemm_util.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 try: diff --git a/megablocks/layers/__init__.py b/megablocks/layers/__init__.py index c52ab00..f0c42de 100644 --- a/megablocks/layers/__init__.py +++ b/megablocks/layers/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from megablocks.layers.dmoe import dMoE diff --git a/megablocks/layers/activation_fn.py b/megablocks/layers/activation_fn.py index 102685f..736d311 100644 --- a/megablocks/layers/activation_fn.py +++ b/megablocks/layers/activation_fn.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from typing import Callable diff --git a/megablocks/layers/all_to_all.py b/megablocks/layers/all_to_all.py index 230f81b..82a6f40 100644 --- a/megablocks/layers/all_to_all.py +++ b/megablocks/layers/all_to_all.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/layers/arguments.py b/megablocks/layers/arguments.py index 08cf3cc..efe131d 100644 --- a/megablocks/layers/arguments.py +++ b/megablocks/layers/arguments.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import dataclasses diff --git a/megablocks/layers/common.py b/megablocks/layers/common.py index 890da35..ee30e79 100644 --- a/megablocks/layers/common.py +++ b/megablocks/layers/common.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/layers/dmlp_registry.py b/megablocks/layers/dmlp_registry.py index ac43035..d765bd0 100644 --- a/megablocks/layers/dmlp_registry.py +++ b/megablocks/layers/dmlp_registry.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from typing import Union diff --git a/megablocks/layers/dmoe.py b/megablocks/layers/dmoe.py index 83f87b8..e683f8a 100644 --- a/megablocks/layers/dmoe.py +++ b/megablocks/layers/dmoe.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/megablocks/layers/gelu.py b/megablocks/layers/gelu.py index da067ac..40b601d 100644 --- a/megablocks/layers/gelu.py +++ b/megablocks/layers/gelu.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import stk diff --git a/megablocks/layers/glu.py b/megablocks/layers/glu.py index ea2102f..fa888a6 100644 --- a/megablocks/layers/glu.py +++ b/megablocks/layers/glu.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import stk diff --git a/megablocks/layers/memory_test.py b/megablocks/layers/memory_test.py index afd3a29..809e317 100644 --- a/megablocks/layers/memory_test.py +++ b/megablocks/layers/memory_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import gc diff --git a/megablocks/layers/mlp.py b/megablocks/layers/mlp.py index abee132..1cae4fb 100644 --- a/megablocks/layers/mlp.py +++ b/megablocks/layers/mlp.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from typing import Any diff --git a/megablocks/layers/moe.py b/megablocks/layers/moe.py index 6114fc1..e5eaaa8 100644 --- a/megablocks/layers/moe.py +++ b/megablocks/layers/moe.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/megablocks/layers/mpu.py b/megablocks/layers/mpu.py index 8b2cd9a..6aa0015 100644 --- a/megablocks/layers/mpu.py +++ b/megablocks/layers/mpu.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/layers/router.py b/megablocks/layers/router.py index 6a356fd..42cfbe1 100644 --- a/megablocks/layers/router.py +++ b/megablocks/layers/router.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/layers/sharedexpert_registry.py b/megablocks/layers/sharedexpert_registry.py index 07ed662..0f62db3 100644 --- a/megablocks/layers/sharedexpert_registry.py +++ b/megablocks/layers/sharedexpert_registry.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from typing import Union diff --git a/megablocks/layers/testing.py b/megablocks/layers/testing.py index d1f12ca..4cd9500 100644 --- a/megablocks/layers/testing.py +++ b/megablocks/layers/testing.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/layers/weight_parallel.py b/megablocks/layers/weight_parallel.py index bbc2571..82effec 100644 --- a/megablocks/layers/weight_parallel.py +++ b/megablocks/layers/weight_parallel.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import stk diff --git a/megablocks/ops/__init__.py b/megablocks/ops/__init__.py index 2951ca1..b9dc286 100644 --- a/megablocks/ops/__init__.py +++ b/megablocks/ops/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from megablocks.ops.binned_gather import binned_gather diff --git a/megablocks/ops/all_to_all_benchmark.py b/megablocks/ops/all_to_all_benchmark.py index 58bfcb9..b3a8537 100644 --- a/megablocks/ops/all_to_all_benchmark.py +++ b/megablocks/ops/all_to_all_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/binned_gather.py b/megablocks/ops/binned_gather.py index cf69f90..8a22317 100644 --- a/megablocks/ops/binned_gather.py +++ b/megablocks/ops/binned_gather.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/binned_scatter.py b/megablocks/ops/binned_scatter.py index dd15ecc..f65fbe8 100644 --- a/megablocks/ops/binned_scatter.py +++ b/megablocks/ops/binned_scatter.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/cumsum.py b/megablocks/ops/cumsum.py index c3804a5..09b23ab 100644 --- a/megablocks/ops/cumsum.py +++ b/megablocks/ops/cumsum.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 # NOTE: Torch needs to be imported before the custom diff --git a/megablocks/ops/gather.py b/megablocks/ops/gather.py index 4f52a95..a335273 100644 --- a/megablocks/ops/gather.py +++ b/megablocks/ops/gather.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/histogram.py b/megablocks/ops/histogram.py index 49b3ba9..7660e82 100644 --- a/megablocks/ops/histogram.py +++ b/megablocks/ops/histogram.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 # NOTE: Torch needs to be imported before the custom diff --git a/megablocks/ops/histogram_benchmark.py b/megablocks/ops/histogram_benchmark.py index 955d264..9de8e65 100644 --- a/megablocks/ops/histogram_benchmark.py +++ b/megablocks/ops/histogram_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import unittest diff --git a/megablocks/ops/matmul_benchmark.py b/megablocks/ops/matmul_benchmark.py index 3b766d1..bfa7b7c 100644 --- a/megablocks/ops/matmul_benchmark.py +++ b/megablocks/ops/matmul_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import unittest diff --git a/megablocks/ops/padded_gather.py b/megablocks/ops/padded_gather.py index 1bca6bd..b57a518 100644 --- a/megablocks/ops/padded_gather.py +++ b/megablocks/ops/padded_gather.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/padded_scatter.py b/megablocks/ops/padded_scatter.py index 8a2c4c7..1ca1605 100644 --- a/megablocks/ops/padded_scatter.py +++ b/megablocks/ops/padded_scatter.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/padded_scatter_benchmark.py b/megablocks/ops/padded_scatter_benchmark.py index 7c11e0a..81dde4e 100644 --- a/megablocks/ops/padded_scatter_benchmark.py +++ b/megablocks/ops/padded_scatter_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import unittest diff --git a/megablocks/ops/permute_benchmark.py b/megablocks/ops/permute_benchmark.py index 11a4201..837f07e 100644 --- a/megablocks/ops/permute_benchmark.py +++ b/megablocks/ops/permute_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import unittest diff --git a/megablocks/ops/repeat.py b/megablocks/ops/repeat.py index 9f02c53..61bb04b 100644 --- a/megablocks/ops/repeat.py +++ b/megablocks/ops/repeat.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 # Copyright 2024 MosaicML MegaBlocks authors diff --git a/megablocks/ops/replicate.py b/megablocks/ops/replicate.py index 3853494..b7cb9c3 100644 --- a/megablocks/ops/replicate.py +++ b/megablocks/ops/replicate.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 # NOTE: Torch needs to be imported before the custom diff --git a/megablocks/ops/round_up.py b/megablocks/ops/round_up.py index 0300fe4..2c59a78 100644 --- a/megablocks/ops/round_up.py +++ b/megablocks/ops/round_up.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/scatter.py b/megablocks/ops/scatter.py index ccffba3..33f051c 100644 --- a/megablocks/ops/scatter.py +++ b/megablocks/ops/scatter.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import torch diff --git a/megablocks/ops/sort.py b/megablocks/ops/sort.py index 37dd13a..12ec8f3 100644 --- a/megablocks/ops/sort.py +++ b/megablocks/ops/sort.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 # NOTE: Torch needs to be imported before the custom diff --git a/megablocks/ops/sort_benchmark.py b/megablocks/ops/sort_benchmark.py index 4a06580..f28e3f2 100644 --- a/megablocks/ops/sort_benchmark.py +++ b/megablocks/ops/sort_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import unittest diff --git a/megablocks/ops/sum.py b/megablocks/ops/sum.py index ac55d24..aa81334 100644 --- a/megablocks/ops/sum.py +++ b/megablocks/ops/sum.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 diff --git a/megablocks/ops/topology.py b/megablocks/ops/topology.py index fb07210..ba4ade0 100644 --- a/megablocks/ops/topology.py +++ b/megablocks/ops/topology.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 # NOTE: Torch needs to be imported before the custom diff --git a/setup.py b/setup.py index 9149352..fa15ee4 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 """MegaBlocks package setup.""" diff --git a/tests/conftest.py b/tests/conftest.py index afe3e43..663bda3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import os diff --git a/tests/fixtures/autouse.py b/tests/fixtures/autouse.py index 986972e..6805f3c 100644 --- a/tests/fixtures/autouse.py +++ b/tests/fixtures/autouse.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import gc diff --git a/tests/fixtures/fixtures.py b/tests/fixtures/fixtures.py index 577c077..4039db7 100644 --- a/tests/fixtures/fixtures.py +++ b/tests/fixtures/fixtures.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import pytest diff --git a/tests/layers/dmoe_test.py b/tests/layers/dmoe_test.py index 5a1e9d5..a737ef4 100644 --- a/tests/layers/dmoe_test.py +++ b/tests/layers/dmoe_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from functools import partial diff --git a/tests/layers/glu_test.py b/tests/layers/glu_test.py index 5ede1db..d89af89 100644 --- a/tests/layers/glu_test.py +++ b/tests/layers/glu_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from functools import partial diff --git a/tests/layers/moe_test.py b/tests/layers/moe_test.py index a4032c2..dd40ef9 100644 --- a/tests/layers/moe_test.py +++ b/tests/layers/moe_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from functools import partial diff --git a/tests/layers/parallelism_test.py b/tests/layers/parallelism_test.py index 6fd4d1e..35e40a0 100644 --- a/tests/layers/parallelism_test.py +++ b/tests/layers/parallelism_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import functools diff --git a/tests/ops/binned_gather_test.py b/tests/ops/binned_gather_test.py index 3906d9c..c165086 100644 --- a/tests/ops/binned_gather_test.py +++ b/tests/ops/binned_gather_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/ops/binned_scatter_test.py b/tests/ops/binned_scatter_test.py index f3c352b..b725700 100644 --- a/tests/ops/binned_scatter_test.py +++ b/tests/ops/binned_scatter_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/ops/cumsum_test.py b/tests/ops/cumsum_test.py index 1e79db1..5d8b082 100644 --- a/tests/ops/cumsum_test.py +++ b/tests/ops/cumsum_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import pytest diff --git a/tests/ops/histogram_test.py b/tests/ops/histogram_test.py index 7a0fc9a..d6d3f23 100644 --- a/tests/ops/histogram_test.py +++ b/tests/ops/histogram_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import pytest diff --git a/tests/ops/padded_gather_test.py b/tests/ops/padded_gather_test.py index a9c880b..7198099 100644 --- a/tests/ops/padded_gather_test.py +++ b/tests/ops/padded_gather_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/ops/padded_scatter_test.py b/tests/ops/padded_scatter_test.py index d4fbba9..0e80dbb 100644 --- a/tests/ops/padded_scatter_test.py +++ b/tests/ops/padded_scatter_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/ops/replicate_test.py b/tests/ops/replicate_test.py index 4076d9d..aeb1405 100644 --- a/tests/ops/replicate_test.py +++ b/tests/ops/replicate_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np diff --git a/tests/ops/sort_test.py b/tests/ops/sort_test.py index 0f86984..147426e 100644 --- a/tests/ops/sort_test.py +++ b/tests/ops/sort_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 from typing import Dict, Optional, Union diff --git a/tests/ops/topology_test.py b/tests/ops/topology_test.py index cbffcbe..dc3c0ae 100644 --- a/tests/ops/topology_test.py +++ b/tests/ops/topology_test.py @@ -1,4 +1,4 @@ -# Copyright 2024 Databricks MegaBlocks authors +# Copyright 2024 Databricks # SPDX-License-Identifier: Apache-2.0 import numpy as np