Emulate i128 ops

LLVM 7.1 (and thus nvvm) lacks native 128-bit integer intrinsics, so emulate them using 64-bit operations.

I also added an example with `sha2` to confirm it works and show
folks that crates.io crates can often be used directly unmodified.

Fixes #207.

Author: LegNeato

.github/workflows/ci_windows.yml

@@ -77,7 +77,7 @@ jobs:
         run: cargo build --all-features -p cust_raw
 
       - name: Build
-        run: cargo build --workspace --exclude "optix*" --exclude "path-tracer" --exclude "denoiser" --exclude "vecadd*" --exclude "gemm*" --exclude "ex*" --exclude "cudnn*"
+        run: cargo build --workspace --exclude "optix*" --exclude "path-tracer" --exclude "denoiser" --exclude "vecadd*" --exclude "gemm*" --exclude "ex*" --exclude "cudnn*" --exclude "sha2*"
 
       # Don't currently test because many tests rely on the system having a CUDA GPU
       # - name: Test
@@ -86,7 +86,7 @@ jobs:
       - name: Check documentation
         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"
+        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 "sha2*" --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

@@ -14,6 +14,8 @@ members = [
   "examples/cuda/gemm/kernels",
   "examples/cuda/path_tracer",
   "examples/cuda/path_tracer/kernels",
+  "examples/cuda/sha2_crates_io",
+  "examples/cuda/sha2_crates_io/kernels",
 
   "examples/optix/*",
   "tests/compiletests",

crates/optix-sys/build/main.rs

@@ -16,7 +16,7 @@ fn main() {
 
     println!("cargo::rerun-if-changed=build");
     for e in sdk.related_optix_envs() {
-        println!("cargo::rerun-if-env-changed={}", e);
+        println!("cargo::rerun-if-env-changed={e}");
     }
     // Emit metadata for the build script.
     println!("cargo::metadata=root={}", sdk.optix_root().display());
@@ -36,7 +36,7 @@ fn main() {
     let metadata_optix_include = env::join_paths(sdk.optix_include_paths())
         .map(|s| s.to_string_lossy().to_string())
         .expect("Failed to build metadata for include.");
-    println!("cargo::metadata=include_dir={}", metadata_optix_include);
+    println!("cargo::metadata=include_dir={metadata_optix_include}");
 
     // Generate optix bindings.
     create_optix_bindings(&sdk, &cuda_include_paths);

crates/optix-sys/build/optix_sdk.rs

@@ -89,7 +89,7 @@ impl OptiXSdk {
             .ok_or("Cannot find OPTIX_VERSION from OptiX header file.")?;
         let version = version
             .parse::<u32>()
-            .map_err(|_| format!("Cannot parse OPTIX_VERSION as u32: '{}'", version))?;
+            .map_err(|_| format!("Cannot parse OPTIX_VERSION as u32: '{version}'"))?;
         Ok(version)
     }
 }

crates/rustc_codegen_nvvm/libintrinsics.bc

@@ -286,6 +286,66 @@ start:
 }
 declare {<2 x i64>, i1} @__rust_u128_mulo(<2 x i64>, <2 x i64>) #0
 
+; Division operations from compiler-builtins
+define <2 x i64> @__nvvm_divti3(<2 x i64>, <2 x i64>) #0 {
+start:
+  %2 = call <2 x i64> @__divti3(<2 x i64> %0, <2 x i64> %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__divti3(<2 x i64>, <2 x i64>) #0
+
+define <2 x i64> @__nvvm_udivti3(<2 x i64>, <2 x i64>) #0 {
+start:
+  %2 = call <2 x i64> @__udivti3(<2 x i64> %0, <2 x i64> %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__udivti3(<2 x i64>, <2 x i64>) #0
+
+; Remainder operations from compiler-builtins
+define <2 x i64> @__nvvm_modti3(<2 x i64>, <2 x i64>) #0 {
+start:
+  %2 = call <2 x i64> @__modti3(<2 x i64> %0, <2 x i64> %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__modti3(<2 x i64>, <2 x i64>) #0
+
+define <2 x i64> @__nvvm_umodti3(<2 x i64>, <2 x i64>) #0 {
+start:
+  %2 = call <2 x i64> @__umodti3(<2 x i64> %0, <2 x i64> %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__umodti3(<2 x i64>, <2 x i64>) #0
+
+; Multiplication from compiler-builtins
+define <2 x i64> @__nvvm_multi3(<2 x i64>, <2 x i64>) #0 {
+start:
+  %2 = call <2 x i64> @__multi3(<2 x i64> %0, <2 x i64> %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__multi3(<2 x i64>, <2 x i64>) #0
+
+; Shift operations from compiler-builtins
+define <2 x i64> @__nvvm_ashlti3(<2 x i64>, i32) #0 {
+start:
+  %2 = call <2 x i64> @__ashlti3(<2 x i64> %0, i32 %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__ashlti3(<2 x i64>, i32) #0
+
+define <2 x i64> @__nvvm_ashrti3(<2 x i64>, i32) #0 {
+start:
+  %2 = call <2 x i64> @__ashrti3(<2 x i64> %0, i32 %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__ashrti3(<2 x i64>, i32) #0
+
+define <2 x i64> @__nvvm_lshrti3(<2 x i64>, i32) #0 {
+start:
+  %2 = call <2 x i64> @__lshrti3(<2 x i64> %0, i32 %1)
+  ret <2 x i64> %2
+}
+declare <2 x i64> @__lshrti3(<2 x i64>, i32) #0
+
 ; Required because we need to explicitly generate { i32, i1 } for the following intrinsics
 ; except rustc will not generate them (it will make { i32, i8 }) which libnvvm rejects.