Add compiletests

These tests check that code compiles (or doesn't) as expected.
The framework also has the capability to check the generated ptx (see
the tests in `dis/`.

These tests are not run, so they do not validate runtime behavior.
This also means they can run on machines without a GPU, like GitHub
Actions runners.

This is largely modeled / structured after rust-gpu's compiletests
in anticipation of a project merge one day.

Author: LegNeato

.cargo/config.toml

@@ -1,2 +1,3 @@
 [alias]
 xtask = "run -p xtask --bin xtask --"
+compiletest = "run --release -p compiletests --"

.github/workflows/ci_linux.yml

@@ -153,3 +153,18 @@ jobs:
           MODAL_TOKEN_SECRET: ${{ secrets.MODAL_TOKEN_SECRET }}
         run: |
           echo "Stubbed out"
+  compiletest:
+    name: Compile tests
+    runs-on: ubuntu-latest
+    container:
+      image: "ghcr.io/rust-gpu/rust-cuda-ubuntu24-cuda12:latest"
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+      - name: Run cargo version
+        run: cargo --version
+      - name: Rustfmt compiletests
+        shell: bash
+        run: shopt -s globstar && rustfmt --check tests/compiletests/ui/**/*.rs
+      - name: Compiletest
+        run: cargo run -p compiletests --release --no-default-features -- --target-arch compute_61,compute_70,compute_90

.github/workflows/ci_windows.yml

@@ -26,7 +26,18 @@ jobs:
             target: x86_64-pc-windows-msvc
             cuda: "12.8.1"
             linux-local-args: []
-            sub-packages: ["nvcc", "nvrtc", "nvrtc_dev", "cuda_profiler_api", "cudart", "cublas", "cublas_dev", "curand", "curand_dev"]
+            sub-packages:
+              [
+                "nvcc",
+                "nvrtc",
+                "nvrtc_dev",
+                "cuda_profiler_api",
+                "cudart",
+                "cublas",
+                "cublas_dev",
+                "curand",
+                "curand_dev",
+              ]
 
     steps:
       - name: Checkout repository
@@ -41,7 +52,7 @@ jobs:
           linux-local-args: ${{ toJson(matrix.linux-local-args) }}
           use-local-cache: false
           sub-packages: ${{ toJson(matrix.sub-packages) }}
-          log-file-suffix: '${{matrix.os}}-${{matrix.cuda}}'
+          log-file-suffix: "${{matrix.os}}-${{matrix.cuda}}"
 
       - name: Verify CUDA installation
         run: nvcc --version
@@ -76,3 +87,6 @@ jobs:
         env:
           RUSTDOCFLAGS: -Dwarnings
         run: cargo doc --workspace --all-features --document-private-items --no-deps --exclude "optix*" --exclude "path-tracer" --exclude "denoiser" --exclude "vecadd*" --exclude "gemm*" --exclude "ex*" --exclude "cudnn*" --exclude "cust_raw"
+      # Disabled due to dll issues, someone with  Windows knowledge needed
+      # - name: Compiletest
+      #  run: cargo run -p compiletests --release --no-default-features -- --target-arch compute_61,compute_70,compute_90

Cargo.toml

@@ -16,6 +16,8 @@ members = [
   "examples/cuda/path_tracer/kernels",
 
   "examples/optix/*",
+  "tests/compiletests",
+  "tests/compiletests/deps-helper",
 ]
 
 exclude = [
@@ -24,3 +26,7 @@ exclude = [
 
 [profile.dev.package.rustc_codegen_nvvm]
 opt-level = 3
+
+[workspace.dependencies]
+cuda_std = { path = "crates/cuda_std" }
+cuda_builder = { path = "crates/cuda_builder" }

crates/rustc_codegen_nvvm/src/context.rs

@@ -557,6 +557,10 @@ pub struct CodegenArgs {
     pub override_libm: bool,
     pub use_constant_memory_space: bool,
     pub final_module_path: Option<PathBuf>,
+    pub disassemble: bool,
+    pub disassemble_fn: Option<String>,
+    pub disassemble_entry: Option<String>,
+    pub disassemble_globals: bool,
 }
 
 impl CodegenArgs {
@@ -569,7 +573,13 @@ impl CodegenArgs {
         // TODO: replace this with a "proper" arg parser.
         let mut cg_args = Self::default();
 
+        let mut skip_next = false;
         for (idx, arg) in args.iter().enumerate() {
+            if skip_next {
+                skip_next = false;
+                continue;
+            }
+
             if let Ok(flag) = NvvmOption::from_str(arg) {
                 cg_args.nvvm_options.push(flag);
             } else if arg == "--override-libm" {
@@ -580,6 +590,29 @@ impl CodegenArgs {
                 cg_args.final_module_path = Some(PathBuf::from(
                     args.get(idx + 1).expect("No path for --final-module-path"),
                 ));
+                skip_next = true;
+            } else if arg == "--disassemble" {
+                cg_args.disassemble = true;
+            } else if arg == "--disassemble-globals" {
+                cg_args.disassemble_globals = true;
+            } else if arg == "--disassemble-fn" {
+                cg_args.disassemble_fn = Some(
+                    args.get(idx + 1)
+                        .expect("No function name for --disassemble-fn")
+                        .clone(),
+                );
+                skip_next = true;
+            } else if let Some(func) = arg.strip_prefix("--disassemble-fn=") {
+                cg_args.disassemble_fn = Some(func.to_string());
+            } else if arg == "--disassemble-entry" {
+                cg_args.disassemble_entry = Some(
+                    args.get(idx + 1)
+                        .expect("No entry name for --disassemble-entry")
+                        .clone(),
+                );
+                skip_next = true;
+            } else if let Some(entry) = arg.strip_prefix("--disassemble-entry=") {
+                cg_args.disassemble_entry = Some(entry.to_string());
             }
         }
 

crates/rustc_codegen_nvvm/src/lib.rs

@@ -52,6 +52,7 @@ mod lto;
 mod mono_item;
 mod nvvm;
 mod override_fns;
+mod ptx_filter;
 mod target;
 mod ty;
 
@@ -216,14 +217,30 @@ impl CodegenBackend for NvvmCodegenBackend {
         let cmdline = sess.opts.cg.target_feature.split(',');
         let cfg = sess.target.options.features.split(',');
 
-        let target_features: Vec<_> = cfg
+        let mut target_features: Vec<_> = cfg
             .chain(cmdline)
             .filter(|l| l.starts_with('+'))
             .map(|l| &l[1..])
             .filter(|l| !l.is_empty())
             .map(rustc_span::Symbol::intern)
             .collect();
 
+        // Add backend-synthesized features (e.g., hierarchical compute capabilities)
+        // Parse CodegenArgs to get the architecture from llvm-args
+        let args = context::CodegenArgs::from_session(sess);
+        for opt in &args.nvvm_options {
+            if let ::nvvm::NvvmOption::Arch(arch) = opt {
+                // Add all features up to and including the current architecture
+                let backend_features = arch.all_target_features();
+                target_features.extend(
+                    backend_features
+                        .iter()
+                        .map(|f| rustc_span::Symbol::intern(f)),
+                );
+                break;
+            }
+        }
+
         // For NVPTX, all target features are stable
         let unstable_target_features = target_features.clone();
 

crates/rustc_codegen_nvvm/src/link.rs

@@ -30,6 +30,7 @@ use tracing::{debug, trace};
 
 use crate::LlvmMod;
 use crate::context::CodegenArgs;
+use crate::ptx_filter::{PtxFilter, PtxFilterConfig};
 
 pub(crate) struct NvvmMetadataLoader;
 
@@ -305,6 +306,34 @@ fn codegen_into_ptx_file(
         }
     };
 
+    // If disassembly is requested, print PTX to stderr
+    if (args.disassemble
+        || args.disassemble_globals
+        || args.disassemble_fn.is_some()
+        || args.disassemble_entry.is_some())
+        && let Ok(ptx_str) = std::str::from_utf8(&ptx_bytes)
+    {
+        let config = PtxFilterConfig::from_codegen_args(&args);
+        let filter = PtxFilter::new(config);
+        let output = filter.filter(ptx_str);
+        if !output.is_empty() {
+            // Check if we're in JSON mode by checking the error format
+            use rustc_session::config::ErrorOutputType;
+            match sess.opts.error_format {
+                ErrorOutputType::Json { .. } => {
+                    sess.dcx()
+                        .err("PTX disassembly output in JSON mode is not supported");
+                }
+                _ => {
+                    // In normal mode, just print to stderr
+                    // Replace tabs with spaces for cleaner output
+                    let output = output.replace('\t', "    ");
+                    eprintln!("{output}");
+                }
+            }
+        }
+    }
+
     std::fs::write(out_filename, ptx_bytes)
 }