[builder] feat: implement upstream field for build.toml.

kernel-builder/Cargo.toml

@@ -23,6 +23,7 @@ serde_json = "1"
 serde-value = "0.7"
 thiserror = "1"
 toml = "0.8"
+url = { version = "2", features = ["serde"] }
 
 [build-dependencies]
 minijinja-embed = "2.5"

kernel-builder/src/config/compat.rs

@@ -5,6 +5,7 @@ use serde_value::Value;
 use super::{v1, v2, v3, Build};
 
 #[derive(Debug)]
+#[allow(clippy::large_enum_variant)]
 pub enum BuildCompat {
     V1(v1::Build),
     V2(v2::Build),

kernel-builder/src/config/mod.rs

@@ -61,6 +61,9 @@ pub struct General {
     /// Hugging Face Hub license identifier.
     pub license: Option<String>,
 
+    /// Source repository or reference for the kernel code.
+    pub upstream: Option<url::Url>,
+
     pub backends: Vec<Backend>,
     pub hub: Option<Hub>,
     pub python_depends: Option<Vec<String>>,

kernel-builder/src/config/v1.rs

@@ -106,6 +106,7 @@ impl TryFrom<Build> for super::Build {
                 name: build.general.name,
                 version: None,
                 license: None,
+                upstream: None,
                 backends,
                 hub: None,
                 neuron: None,

kernel-builder/src/config/v2.rs

@@ -171,6 +171,7 @@ impl General {
             name: general.name,
             version: None,
             license: None,
+            upstream: None,
             backends,
             cuda,
             hub: general.hub.map(Into::into),

kernel-builder/src/config/v3.rs

@@ -38,6 +38,8 @@ pub struct General {
 
     pub license: Option<String>,
 
+    pub upstream: Option<url::Url>,
+
     pub backends: Vec<Backend>,
 
     pub cuda: Option<CudaGeneral>,
@@ -183,6 +185,7 @@ impl From<General> for super::General {
             name: general.name,
             version: general.version,
             license: general.license,
+            upstream: general.upstream,
             backends: general.backends.into_iter().map(Into::into).collect(),
             cuda: general.cuda.map(Into::into),
             hub: general.hub.map(Into::into),
@@ -363,6 +366,7 @@ impl From<super::General> for General {
             name: general.name,
             version: general.version,
             license: general.license,
+            upstream: general.upstream,
             backends: general.backends.into_iter().map(Into::into).collect(),
             cuda: general.cuda.map(Into::into),
             hub: general.hub.map(Into::into),

kernel-builder/src/pyproject/common.rs

@@ -34,6 +34,7 @@ pub fn write_metadata(general: &General, file_set: &mut FileSet) -> Result<()> {
         let metadata = Metadata {
             version: general.version,
             license: general.license.clone(),
+            upstream: general.upstream.clone(),
             python_depends,
         };
 

kernel-builder/src/pyproject/metadata.rs

@@ -7,5 +7,7 @@ pub struct Metadata {
     pub version: Option<usize>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub license: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub upstream: Option<url::Url>,
     pub python_depends: Vec<String>,
 }

template/build.toml

@@ -8,6 +8,7 @@ backends = [
 ]
 name = "__KERNEL_NAME__"
 version = 1
+upstream = "__UPSTREAM_URL__"
 
 [torch]
 src = [