From 2d7e837e3f731b0d61685e60cd4dab51dc1f8f00 Mon Sep 17 00:00:00 2001
From: Tom Hennigan <tomhennigan@google.com>
Date: Sun, 19 May 2024 11:57:20 -0700
Subject: [PATCH] Add default applicable licenses to JMP.

PiperOrigin-RevId: 635262393
---
 jmp/BUILD      | 11 ++++++++++-
 jmp/_src/BUILD |  5 ++++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/jmp/BUILD b/jmp/BUILD
index 000cac2..4d8a770 100644
--- a/jmp/BUILD
+++ b/jmp/BUILD
@@ -1,7 +1,16 @@
 load("//third_party/bazel_rules/rules_python/python:py_library.bzl", "py_library")
+load("//tools/build_defs/license:license.bzl", "license")
 
 # Description: JMP is a JAX Mixed Precision library.
-package(default_visibility = ["//visibility:private"])
+package(
+    default_applicable_licenses = [":license"],
+    default_visibility = ["//visibility:private"],
+)
+
+license(
+    name = "license",
+    package_name = "jmp",
+)
 
 licenses(["notice"])
 
diff --git a/jmp/_src/BUILD b/jmp/_src/BUILD
index abf2c3b..f6ce549 100644
--- a/jmp/_src/BUILD
+++ b/jmp/_src/BUILD
@@ -1,6 +1,9 @@
 load("//jmp/_src:build_defs.bzl", "jmp_py_library", "jmp_py_test")
 
-package(default_visibility = ["//jmp:__subpackages__"])
+package(
+    default_applicable_licenses = ["//jmp:license"],
+    default_visibility = ["//jmp:__subpackages__"],
+)
 
 licenses(["notice"])