Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 12 additions & 7 deletions .github/workflows/ci-gpu.yml
Original file line number Diff line number Diff line change
@@ -1,12 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction,
# disclosure or distribution of this material and related documentation
# without an express license agreement from NVIDIA CORPORATION or
# its affiliates is strictly prohibited.
# 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.

# GPU tests live in a separate workflow because NVIDIA self-hosted runners
# block pull_request events entirely. Keeping them here avoids a confusing
Expand Down
19 changes: 12 additions & 7 deletions .github/workflows/long-running-tests.yml
Original file line number Diff line number Diff line change
@@ -1,12 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction,
# disclosure or distribution of this material and related documentation
# without an express license agreement from NVIDIA CORPORATION or
# its affiliates is strictly prohibited.
# 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.

# Long-running GPU tests (20+ min each): statistical validation,
# multi-orientation inference, LER regression, and full-epoch training.
Expand Down
19 changes: 12 additions & 7 deletions .github/workflows/pr_file_check.yaml
Original file line number Diff line number Diff line change
@@ -1,12 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction,
# disclosure or distribution of this material and related documentation
# without an express license agreement from NVIDIA CORPORATION or
# its affiliates is strictly prohibited.
# 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.
name: Check for Large Files and Restricted Extensions

on:
Expand Down
70 changes: 0 additions & 70 deletions Contributor_License_Agreement.md

This file was deleted.

14 changes: 13 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# 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.
#
# Pre-decoder training image.
#
Expand Down
1 change: 1 addition & 0 deletions LICENSE
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.

Apache License
Version 2.0, January 2004
Expand Down
20 changes: 17 additions & 3 deletions code/scripts/spdx_headers.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@

# Old proprietary license bodies (year-agnostic, used for replacement)
_OLD_HASH_BODY = (
"# SPDX-License-Identifier: Apache-2.0\n"
"# SPDX-License-Identifier: LicenseRef-NvidiaProprietary\n"
"#\n"
"# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual\n"
"# property and proprietary rights in and to this material, related\n"
Expand Down Expand Up @@ -226,6 +226,11 @@ def _has_header(content: str) -> bool:
return "SPDX-FileCopyrightText" in prefix


def _has_apache_license(content: str) -> bool:
prefix = "\n".join(content.splitlines()[:20])
return "SPDX-License-Identifier: Apache-2.0" in prefix


def _apply_header(path: Path, header: str, style: str) -> bool:
raw = path.read_text(encoding="utf-8")

Expand Down Expand Up @@ -259,7 +264,9 @@ def _apply_header(path: Path, header: str, style: str) -> bool:
def main() -> int:
parser = argparse.ArgumentParser(description="Add or check SPDX headers.")
parser.add_argument(
"--check", action="store_true", help="Fail if any files are missing headers."
"--check",
action="store_true",
help="Fail if any files are missing headers or use a non-Apache-2.0 license.",
)
parser.add_argument(
"--root",
Expand All @@ -271,6 +278,7 @@ def main() -> int:

year = dt.datetime.now().year
missing = []
wrong_license = []
updated = []

for path in sorted(_iter_files(args.root)):
Expand All @@ -282,6 +290,8 @@ def main() -> int:
raw = path.read_text(encoding="utf-8")
if not _has_header(raw):
missing.append(path.relative_to(args.root))
elif not _has_apache_license(raw):
wrong_license.append(path.relative_to(args.root))
else:
if _apply_header(path, header, style):
updated.append(path.relative_to(args.root))
Expand All @@ -290,8 +300,12 @@ def main() -> int:
if missing:
missing_str = "\n".join(f"- {path}" for path in missing)
print("Missing SPDX headers:\n" + missing_str)
if wrong_license:
wrong_str = "\n".join(f"- {path}" for path in wrong_license)
print("SPDX headers without 'SPDX-License-Identifier: Apache-2.0':\n" + wrong_str)
if missing or wrong_license:
return 1
print("All checked files have SPDX headers.")
print("All checked files have Apache-2.0 SPDX headers.")
return 0

if updated:
Expand Down
19 changes: 12 additions & 7 deletions code/tests/test_dataloader_multiprocessing.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction,
# disclosure or distribution of this material and related documentation
# without an express license agreement from NVIDIA CORPORATION or
# its affiliates is strictly prohibited.
# 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.
"""
Multi-worker DataLoader tests for the Stim inference datapipe.

Expand Down
14 changes: 13 additions & 1 deletion conf/config_pre_decoder_memory_surface_model_1_d9.yaml
Original file line number Diff line number Diff line change
@@ -1,5 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# 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.

exp_tag: pre-decoder # The shell script always overrides this with "${EXPERIMENT_NAME}"
output: outputs/${exp_tag} # Main output directory.
Expand Down
14 changes: 13 additions & 1 deletion conf/config_qec_decoder_r13_fp8.yaml
Original file line number Diff line number Diff line change
@@ -1,5 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# 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.
#
# Config for: Ising-Decoder-SurfaceCode-1-Accurate (R=13), aka Model 4 from the corresponding paper, depolarizing p=0.006, experiment qec-decoder-depolarizing-r13-fp8.
# Training uses receptive field 13; evaluation targets distance/n_rounds 13.
Expand Down
14 changes: 13 additions & 1 deletion conf/config_qec_decoder_r9_fp8.yaml
Original file line number Diff line number Diff line change
@@ -1,5 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# 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.
#
# Config for: Ising-Decoder-SurfaceCode-1-Fast (R=9), aka Model 1 from the corresponding paper, depolarizing p=0.006, experiment qec-decoder-depolarizing-r9-fp8.
# Training uses receptive field 9; evaluation targets distance/n_rounds 9.
Expand Down
19 changes: 12 additions & 7 deletions conf/config_v3_timing_test.yaml
Original file line number Diff line number Diff line change
@@ -1,12 +1,17 @@
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
# SPDX-License-Identifier: Apache-2.0
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction,
# disclosure or distribution of this material and related documentation
# without an express license agreement from NVIDIA CORPORATION or
# its affiliates is strictly prohibited.
# 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.

# Timing check config for v3 (Torch HE) branch.
# Timing test: d=5, n_rounds=5, 25-param noise model (p=0.006).
Expand Down
Loading