Skip to content

Commit

Permalink
fix license
Browse files Browse the repository at this point in the history
  • Loading branch information
Eitan Turok committed Aug 8, 2024
1 parent b1ab8b7 commit 3b342ff
Show file tree
Hide file tree
Showing 62 changed files with 62 additions and 62 deletions.
2 changes: 1 addition & 1 deletion .pre-commit/FILE_HEADER
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
Copyright 2024 Databricks MegaBlocks authors
Copyright 2024 Databricks
SPDX-License-Identifier: Apache-2.0
2 changes: 1 addition & 1 deletion megablocks/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from megablocks.layers.arguments import Arguments
Expand Down
2 changes: 1 addition & 1 deletion megablocks/_version.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

"""The MegaBlocks Version."""
Expand Down
2 changes: 1 addition & 1 deletion megablocks/backend/__init__.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0
2 changes: 1 addition & 1 deletion megablocks/backend/kernels.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/benchmark_util.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion megablocks/grouped_gemm_util.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

try:
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from megablocks.layers.dmoe import dMoE
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/activation_fn.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from typing import Callable
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/all_to_all.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/arguments.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import dataclasses
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/common.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/dmlp_registry.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from typing import Union
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/dmoe.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/gelu.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import stk
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/glu.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import stk
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/memory_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import gc
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/mlp.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from typing import Any
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/moe.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/mpu.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/router.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/sharedexpert_registry.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from typing import Union
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/testing.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/layers/weight_parallel.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import stk
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/__init__.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/all_to_all_benchmark.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/binned_gather.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/binned_scatter.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/cumsum.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/gather.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/histogram.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/histogram_benchmark.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import unittest
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/matmul_benchmark.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import unittest
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/padded_gather.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/padded_scatter.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/padded_scatter_benchmark.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import unittest
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/permute_benchmark.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import unittest
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/repeat.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

# Copyright 2024 MosaicML MegaBlocks authors
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/replicate.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/round_up.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/scatter.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import torch
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/sort.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/sort_benchmark.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import unittest
Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/sum.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0


Expand Down
2 changes: 1 addition & 1 deletion megablocks/ops/topology.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

"""MegaBlocks package setup."""
Expand Down
2 changes: 1 addition & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import os
Expand Down
2 changes: 1 addition & 1 deletion tests/fixtures/autouse.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import gc
Expand Down
2 changes: 1 addition & 1 deletion tests/fixtures/fixtures.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import pytest
Expand Down
2 changes: 1 addition & 1 deletion tests/layers/dmoe_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from functools import partial
Expand Down
2 changes: 1 addition & 1 deletion tests/layers/glu_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from functools import partial
Expand Down
2 changes: 1 addition & 1 deletion tests/layers/moe_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from functools import partial
Expand Down
2 changes: 1 addition & 1 deletion tests/layers/parallelism_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import functools
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/binned_gather_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/binned_scatter_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/cumsum_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import pytest
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/histogram_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import pytest
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/padded_gather_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/padded_scatter_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/replicate_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/sort_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

from typing import Dict, Optional, Union
Expand Down
2 changes: 1 addition & 1 deletion tests/ops/topology_test.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2024 Databricks MegaBlocks authors
# Copyright 2024 Databricks
# SPDX-License-Identifier: Apache-2.0

import numpy as np
Expand Down

0 comments on commit 3b342ff

Please sign in to comment.