Rollup merge of #150444 - Sa4dUs:offload-intrinsic2, r=ZuseZ4

JonathanBrouwer · web-flow · commit 95b0544fff8c · 2026-01-02T19:00:15.000+01:00
Expose kernel launch options as offload intrinsic args

Allows modifying the workgroup and thread grid dimensions directly from the intrinsic call.

```rust
core::intrinsics::offload(_kernel_1, [256, 1, 1], [32, 1, 1], (x,))
```

r? `@ZuseZ4`
diff --git a/src/offload/usage.md b/src/offload/usage.md
@@ -57,7 +57,7 @@ fn main() {
 
 #[inline(never)]
 unsafe fn kernel(x: *mut [f64; 256]) {
-    core::intrinsics::offload(kernel_1, (x,))
+    core::intrinsics::offload(_kernel_1, [256, 1, 1], [32, 1, 1], (x,))
 }
 
 #[cfg(target_os = "linux")]

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ fn main() {`
`57`	`57`
`58`	`58`	`#[inline(never)]`
`59`	`59`	`unsafe fn kernel(x: *mut [f64; 256]) {`
`60`		`- core::intrinsics::offload(kernel_1, (x,))`
	`60`	`+ core::intrinsics::offload(_kernel_1, [256, 1, 1], [32, 1, 1], (x,))`
`61`	`61`	`}`
`62`	`62`
`63`	`63`	`#[cfg(target_os = "linux")]`