[Bug] Reuse VkInstance across init/reset cycles#465
Conversation
The NVIDIA Vulkan driver has a bug where repeated vkDestroyInstance/vkCreateInstance cycles corrupt SubgroupLocalInvocationId after ~11 cycles. The corruption is permanent for the process lifetime and cannot be recovered from, even with VK_EXT_subgroup_size_control + requiredSubgroupSize=32. Fix: keep the VkInstance alive in VulkanLoader (process-lifetime singleton) and reuse it across qd.init/qd.reset cycles. Only the VkDevice is destroyed and recreated. Verified: 30 consecutive Vulkan init/reset cycles all produce correct SubgroupLocalInvocationId = [0..15]. Reproducer: standalone C program (tmp/vk_subgroup_repro.c) confirms the bug is in the NVIDIA driver, not in Quadrants code.
| // Reuse the VkInstance from a previous init/reset cycle if available. | ||
| // Repeated vkDestroyInstance/vkCreateInstance triggers an NVIDIA driver bug | ||
| // that corrupts SubgroupLocalInvocationId after ~11 cycles. | ||
| VkInstance existing = VulkanLoader::instance().get_instance(); |
There was a problem hiding this comment.
why does VulkanLoader::instance().get_instance(); always return the same instance?
There was a problem hiding this comment.
opus response:
VulkanLoader is a Meyer's singleton — the static local variable ensures there's exactly one instance per process:
quadrants/quadrants/rhi/vulkan/vulkan_loader.h lines 296-299
static VulkanLoader &instance() {
static VulkanLoader instance;
return instance;
}
And it holds the VkInstance as a member:
quadrants/quadrants/rhi/vulkan/vulkan_loader.h lines 44-44
VkInstance vulkan_instance_{VK_NULL_HANDLE};
The flow across qd.init()/qd.reset() cycles:
- First
qd.init(): get_instance() returns VK_NULL_HANDLE, so create_instance() creates a new VkInstance and stores it via load_instance(instance_). qd.reset(): ~VulkanDeviceCreator destroys the VkDevice but no longer destroys the VkInstance. The singleton's vulkan_instance_ still points to the live handle.- Second
qd.init(): get_instance() returns the same live VkInstance, so create_instance() reuses it and returns early.
The VulkanLoader singleton outlives all VulkanDeviceCreator instances — it's destroyed only at process exit.
There was a problem hiding this comment.
why do we have both instance_ and vulkan_instance_?
There was a problem hiding this comment.
opus:
They're in different classes with different lifetimes:
• VulkanDeviceCreator::instance_ — a local copy on the creator object, used during device setup (picking physical device, creating logical device, etc.). It's created per
qd.init() call and dies with the creator on qd.reset().
• VulkanLoader::vulkan_instance_ — the singleton's copy that persists for the process lifetime. It's also used by load_function() to resolve Vulkan proc addresses.
They point to the same VkInstance handle. It's redundant — instance_ could just read from the loader each time. But it's existing code structure: VulkanDeviceCreator was
originally designed to own and destroy its VkInstance, so it kept its own copy. Our fix just made it stop destroying it and start reusing the loader's copy on subsequent
cycles.
Cleaning it up (removing instance_ and always going through the loader) would be a nice follow-up refactor, but not strictly necessary for the fix.
There was a problem hiding this comment.
how much code change would be required to clean up instance_ ? are there any perofrmanace implications?
There was a problem hiding this comment.
Opus:
11 uses of instance_ in the file. To remove it you'd:
- Remove VkInstance instance_ from the header (1 line)
- Replace all 11 occurrences with VulkanLoader::instance().get_instance() (or cache it in a local variable at the top of each method)
That's ~15 lines changed. No performance implications — VulkanLoader::instance() is a static local reference (zero-cost after first call), and get_instance() is an inline
getter returning a member. These are all called during init/teardown, never in hot paths.
It's a trivial cleanup but touches many lines for no functional change, so it's a judgment call whether it's worth the diff noise.
| } | ||
| vkDestroyDevice(device_, kNoVkAllocCallbacks); | ||
| vkDestroyInstance(instance_, kNoVkAllocCallbacks); | ||
| // VkInstance is intentionally kept alive in VulkanLoader (process-lifetime). |
There was a problem hiding this comment.
note to self: method VulkanDeviceCreator::~VulkanDeviceCreator()
| // Repeated vkDestroyInstance/vkCreateInstance triggers an NVIDIA driver bug | ||
| // that corrupts SubgroupLocalInvocationId after ~11 cycles. | ||
| VkInstance existing = VulkanLoader::instance().get_instance(); | ||
| if (existing != VK_NULL_HANDLE) { |
There was a problem hiding this comment.
what happens if existing is null?
There was a problem hiding this comment.
Opus:
If it's null, we fall through to the original code below that creates a new VkInstance via vkCreateInstance(...) and stores it in the loader via
VulkanLoader::instance().load_instance(instance_) at line 419.
So it's just the first-call vs subsequent-call path:
• First qd.init(): existing == VK_NULL_HANDLE → create new VkInstance, store in loader
• Subsequent qd.init(): existing != VK_NULL_HANDLE → reuse it, return early
| @@ -78,6 +78,7 @@ bool VulkanLoader::check_vulkan_device() { | |||
|
|
|||
| if (instance) { | |||
There was a problem hiding this comment.
what si the purpose of this code?
There was a problem hiding this comment.
Opus:
That's in VulkanLoader::check_vulkan_device() — a probe function that runs once during VulkanLoader::init() to check if there's a GPU with a compute queue. It creates a
temporary VkInstance just to enumerate physical devices, then destroys it.
The problem was that load_instance(instance) on line 46 stored the temporary handle into vulkan_instance_, and then line 80 destroyed it — leaving vulkan_instance_ as a
dangling pointer. Without the vulkan_instance_ = VK_NULL_HANDLE fix, our reuse check in create_instance() would see a non-null handle and try to reuse a destroyed instance.
So that one line I added just ensures the probe cleanup doesn't leave a stale pointer that would fool the reuse logic.
|
to Opus: does this pr have any tests? |
|
Opus: No, the branch has no committed tests. The test I ran (tmp/test_vulkan_instance_reuse.py) was ad-hoc and not committed. |
fe82267 to
6f5d156
Compare
6f5d156 to
e21da70
Compare
|
I have read every line added in this PR, and reviewed the lines. I take responsibilty for the lines added and removed in this PR, and won't blame any issues on Opus. |
…cles Cycles qd.init(vulkan)/qd.reset() 20 times and verifies that SubgroupLocalInvocationId stays stable (same values as first cycle). Vendor-agnostic: works regardless of the hardware's native subgroup size. Without the VkInstance reuse fix, this fails at cycle ~11 on NVIDIA due to a driver bug.
e21da70 to
ce8603b
Compare
There was a problem hiding this comment.
💡 Codex Review
Here are some automated review suggestions for this pull request.
Reviewed commit: c3819b64d6
ℹ️ About Codex in GitHub
Codex has been enabled to automatically review pull requests in this repo. Reviews are triggered when you
- Open a pull request for review
- Mark a draft as ready
- Comment "@codex review".
If Codex has suggestions, it will comment; otherwise it will react with 👍.
When you sign up for Codex through ChatGPT, Codex can also answer questions or update the PR, like "@codex address that feedback".
| if (existing != VK_NULL_HANDLE) { | ||
| instance_ = existing; | ||
| ti_device_->vk_caps().vk_api_version = vk_api_version; | ||
| return; |
There was a problem hiding this comment.
Preserve validation-layer settings when reusing VkInstance
This early return skips all instance creation options, so after one non-debug Vulkan init, later qd.init(..., debug=True) runs (wired via VulkanProgramImpl::materialize_runtime) silently reuse an instance that was created without validation layers. That means debug-mode validation and debug printf behavior are no longer actually enabled after reset/reinit cycles, even though callers explicitly requested them, which can hide correctness issues and make debug behavior depend on init order.
Useful? React with 👍 / 👎.
| // that corrupts SubgroupLocalInvocationId after ~11 cycles. | ||
| VkInstance existing = VulkanLoader::instance().get_instance(); | ||
| if (existing != VK_NULL_HANDLE) { | ||
| instance_ = existing; | ||
| ti_device_->vk_caps().vk_api_version = vk_api_version; | ||
| return; | ||
| } | ||
|
|
||
| VkApplicationInfo app_info{}; | ||
| app_info.sType = VK_STRUCTURE_TYPE_APPLICATION_INFO; |
There was a problem hiding this comment.
🔴 The early-return path added in create_instance() (lines 304-307) only sets instance_ and vk_api_version on the freshly-constructed VulkanDevice, but never sets physical_device_features2 (or surface). Because ti_device_ is rebuilt as std::make_unique() on every qd.init() call, those flags default to false on the 2nd+ cycle. In create_logical_device() the entire if (ti_device_->vk_caps().physical_device_features2) block is the only place that queries VkPhysicalDeviceFeatures2 and enables spirv_has_variable_ptr, spirv_has_float16, spirv_has_int8, spirv_has_atomic_float_add, spirv_has_shader_clock, buffer device address, and 8/16-bit storage — it is skipped entirely on every reinit. Fix: in the early-return path also propagate physical_device_features2 (and surface) from the first-cycle detection, or unconditionally set physical_device_features2 = true since the extension is already loaded into the live VkInstance.
Extended reasoning...
Root cause
create_instance() detects the VK_KHR_get_physical_device_properties2 instance extension inside the vkEnumerateInstanceExtensionProperties loop and sets ti_device_->vk_caps().physical_device_features2 = true (line 381). The new early-return path (lines 304-307) exits the function before that loop, so on the 2nd+ qd.init() call the flag stays at its default value of false. The same loop is also the only place ti_device_->vk_caps().surface is set (line 378), so that cap is silently absent on reinit as well.
Why the freshly-constructed VulkanDevice does not help
VulkanDeviceCreator creates ti_device_ = std::make_unique() in its constructor on every qd.init() cycle. That new VulkanDevice starts with a zero-initialized VulkanCapabilities struct; there is no copy from any previous cycle. So every reinit begins with physical_device_features2 = false regardless of what was detected before.
Code path that is silently skipped
In create_logical_device() at line 724:
if (ti_device_->vk_caps().physical_device_features2) {
// VkPhysicalDeviceFeatures2KHR query block — sole call site
// Sets: spirv_has_variable_ptr, spirv_has_atomic_float_add,
// spirv_has_atomic_float, spirv_has_atomic_float64*,
// spirv_has_float16, spirv_has_int8, spirv_has_atomic_float16*,
// spirv_has_shader_clock, buffer device address,
// 8-bit / 16-bit storage features
}
If this block is skipped, all those DeviceCapability flags remain unset even though the VkDevice is created with those extensions enabled.
Concrete step-by-step proof
- First qd.init(vulkan): get_instance() returns VK_NULL_HANDLE, falls through to the full path, enumeration loop runs, physical_device_features2 = true, create_logical_device() sees true, all caps populated correctly.
- qd.reset(): VulkanDeviceCreator is destroyed; VkDevice is destroyed; VkInstance is kept alive in the singleton.
- Second qd.init(vulkan): new VulkanDeviceCreator, new VulkanDevice (physical_device_features2 = false), get_instance() returns the live handle, early return fires, physical_device_features2 is never set, create_logical_device() sees false, entire Features2 block is skipped, spirv_has_float16, spirv_has_int8, spirv_has_atomic_float_add, spirv_has_variable_ptr, spirv_has_shader_clock all remain false.
Impact
Every qd.init() after the first qd.reset() silently degrades the device capability set, excluding float16, int8, atomic float, variable pointers, shader clock, and buffer device address even though the GPU and driver fully support them.
Fix
In the early-return path, set physical_device_features2 = true unconditionally (the extension is already baked into the live VkInstance) and set surface similarly. Alternatively, cache the first-cycle caps in the VulkanLoader singleton and copy them on reuse.
Move instance extension discovery (physical_device_features2, surface) before the VkInstance reuse early-return. Since ti_device_ is freshly constructed on every qd.init() cycle, these caps must be re-detected each time — otherwise f16, i8, atomic float, variable pointers, etc. are silently disabled on the 2nd+ init cycle. Fixes the CI failures flagged by Claude on PR #465.
|
|
||
| void VulkanDeviceCreator::create_instance(uint32_t vk_api_version, | ||
| bool manual_create) { | ||
| // Discover instance extensions and set capability flags on ti_device_. |
There was a problem hiding this comment.
walk me through what happens in create_instance
There was a problem hiding this comment.
Opus:
Here's the flow:
Phase 1 — Capability discovery (lines 300–317)
Enumerates instance extensions available on the system and sets two flags on ti_device_->vk_caps():
• surface — if VK_KHR_surface is present
• physical_device_features2 — if VK_KHR_get_physical_device_properties2 is present
This runs every cycle because ti_device_ is freshly constructed each time (all caps default to false). These flags are critical — physical_device_features2 gates whether
create_logical_device() will query and enable f16, i8, atomic float, variable pointers, shader clock, buffer device address, etc.
Phase 2 — VkInstance reuse check (lines 319–327)
Asks the VulkanLoader singleton for an existing VkInstance. If one exists (2nd+ qd.init() cycle), copies it into instance_, sets the API version, and returns early. This
avoids the NVIDIA driver bug that corrupts subgroup IDs after ~11 vkDestroyInstance/vkCreateInstance cycles.
Phase 3 — First-time VkInstance creation (lines 329–423)
Only runs on the very first qd.init():
- App info (329–335) — standard Vulkan boilerplate
- Validation layers (341–362) — if params_.enable_validation_layer, checks layer support and chains debug messenger into pNext
- Debug printf (364–374) — enables VK_VALIDATION_FEATURE_ENABLE_DEBUG_PRINTF_EXT if validation is on
- Extension set (376–393) — collects required extensions + any additional ones from params_, then adds supported well-known extensions (surface, properties2, external
memory/semaphore, debug utils) - vkCreateInstance (404–405) — creates the actual instance
- Fallback (407–414) — if the driver returns VK_ERROR_INCOMPATIBLE_DRIVER, retries with Vulkan 1.0 API version
- Store in singleton (423) — VulkanLoader::instance().load_instance(instance_) so future cycles can reuse it
There was a problem hiding this comment.
Add a doc string wiht this information
There was a problem hiding this comment.
what is the relationship between ti_device_ and VkInstance?
There was a problem hiding this comment.
They're at different levels of abstraction:
• VkInstance (line 82, instance_) is a raw Vulkan handle — the connection to the Vulkan runtime/driver. It's used to enumerate physical devices, create logical devices, and
resolve function pointers. There's one per process (reused via the VulkanLoader singleton).
• ti_device_ (line 91) is a std::unique_ptr — Quadrants' own wrapper that holds the logical device (VkDevice), capability flags (vk_caps()), memory allocator,
pipeline cache, etc. It's the object that the rest of the Quadrants runtime interacts with.
The relationship: VulkanDeviceCreator uses instance_ during setup to discover what the hardware supports, then populates ti_device_'s capability flags and creates the VkDevice
inside it. Once construction is done, ti_device_ is handed off (via device()) and instance_ is no longer needed by the creator.
Key lifecycle difference: ti_device_ is recreated every qd.init() cycle (fresh VulkanDevice with all caps defaulting to false), while instance_ is now reused across cycles via
the singleton. That's exactly why the capability discovery had to be moved before the early return — the instance persists but the device object doesn't.
|
Thanks! |
# Conflicts: # quadrants/rhi/vulkan/vulkan_device_creator.cpp
* [Misc] Warn user to disable caching when print_ir/QD_DUMP_IR enabled (Genesis-Embodied-AI#425) Co-authored-by: v01dxyz <v01dxyz@v01d.xyz> * [Build] Pin torch version to CUDA 12.8 for CUDA tests (Genesis-Embodied-AI#428) * [Misc] Fixing up taichi-dev urls (Genesis-Embodied-AI#429) * [Perf] Rename cuda_graph to gpu_graph across the codebase (Genesis-Embodied-AI#430) * Misc: fix typo integeral -> integral (Genesis-Embodied-AI#434) Co-authored-by: v01dxyz <v01dxyz@v01d.xyz> * [Perf] CUDA graph 4: call from multiple locations (Genesis-Embodied-AI#420) * [Bug] Fix fastcache not restoring graph_do_while_arg (Genesis-Embodied-AI#435) * [Perf] Cache last-call result in perf_dispatch for single-compatible case (Genesis-Embodied-AI#438) * Fix gpu_graph fallback on old Nvidia GPU. (Genesis-Embodied-AI#443) * Fix shared memory offset not reset between CUDA kernels. (Genesis-Embodied-AI#442) * [Misc] Allow disabling GPU graph via QD_GPU_GRAPH=0 env var (Genesis-Embodied-AI#439) * [Misc] Add named top-level loops (Genesis-Embodied-AI#440) * [Misc] Rename gpu_graph to graph (Genesis-Embodied-AI#446) * [Misc] Add cross-platform shuffle (Genesis-Embodied-AI#447) * [Bug] Fix graph_do_while on Windows: search for cudadevrt.lib (Genesis-Embodied-AI#456) * [Bug] Also search default CUDA toolkit install location on Windows (Genesis-Embodied-AI#461) * [SPIRV] Feature Parity Atomics & Shared Array (Genesis-Embodied-AI#432) * [Misc] Change clang format to 120 characters (Genesis-Embodied-AI#463) * [Misc] CUDA graph 5 Add fatbin (Genesis-Embodied-AI#464) * [Bug] Reuse VkInstance across init/reset cycles (Genesis-Embodied-AI#465) * [Perf] Tiles 1: _load, _store, _eye_ (Genesis-Embodied-AI#466) * [Misc] Remove dead InternalFuncStmt type_check override (Genesis-Embodied-AI#471) * [Perf] Tiles 2: add cholesky and ger (Genesis-Embodied-AI#472) * [Perf] Tiles 2b: add triangular solve (Genesis-Embodied-AI#474) * [Misc] Refactor: use _get_col/_set_col in tiles load/store/init (Genesis-Embodied-AI#475) * [Build] Fix flaky test_clock_accuracy (Genesis-Embodied-AI#436) * Fix AARCH64 emitting invalid asm in CUDA kernels. (Genesis-Embodied-AI#473) Co-authored-by: Hugh Perkins <hughperkins@gmail.com> * [AMDGPU] Enable HIP memory pool and surface pool-exhaustion errors. (Genesis-Embodied-AI#485) * [AMDGPU] Scope hsaco tmp dir per-user to avoid collisions. (Genesis-Embodied-AI#484) * [Perf] Tiles 3: Add slice syntax, qd.outer() and initial doc (Genesis-Embodied-AI#477) * [AMDGPU] Fix gradient computation. (Genesis-Embodied-AI#486) * Enable all backends that are supported in unit tests. (Genesis-Embodied-AI#488) * Fix SPIRV ID overflow for large kernels due to autodiff. (Genesis-Embodied-AI#489) * [Misc] Fix purity checker to allow accessing constants from quadrants modules (Genesis-Embodied-AI#487) * [Misc] Increase tolerance for clock monotonic test (Genesis-Embodied-AI#492) * [CI] Serialize api doc workflow (Genesis-Embodied-AI#494) * [CI] Increase tolerance for clock test (Genesis-Embodied-AI#506) * [CI] Increase clock test tolerance to 20% (Genesis-Embodied-AI#509) * [Perf] Add tensor_type parametrization to tile16 tests (Genesis-Embodied-AI#504) * [Perf] Tiles 4b: Migrate tiles16 tests to enable fastcache (Genesis-Embodied-AI#505) * [Perf] Tiles 4c: add Tiles16x16 proxy (Genesis-Embodied-AI#507) * [Perf] Tiles 4d: Consolidate slice error tests using parametrize (Genesis-Embodied-AI#508) * [Perf] Tiles 4: add SharedArray slice support (Genesis-Embodied-AI#482) * [Perf] Tiles 5: add Cholesky benchmark demo (Genesis-Embodied-AI#483) * [Doc] Add user guide page for subgroup shuffle (Genesis-Embodied-AI#512) * [Perf] Implement cross-platform shuffle_down (Genesis-Embodied-AI#510) * [Perf] Add portable subgroup reduce_add and reduce_all_add (Genesis-Embodied-AI#511) * [Perf] Add first warmup config to perf dispatch (Genesis-Embodied-AI#422) * [AutoDiff] Autodiff 1: Add baseline adstack regression test for unary_collections (Genesis-Embodied-AI#500) * [AutoDiff] Autodiff 2: Implement derivative for tan (Genesis-Embodied-AI#501) * [AutoDiff] Autodiff 3: Recompute tanh/exp on the operand in the reverse pass (Genesis-Embodied-AI#502) * [AutoDiff] Autodiff 4: Mark rsqrt as non-linear for adstack promotion (Genesis-Embodied-AI#503) * [AutoDiff] Autodiff 5: Fix adjoint-alloca placement for GlobalLoads outside the current range-for (Genesis-Embodied-AI#496) * [AutoDiff] Autodiff 6: Adstack regression tests (Genesis-Embodied-AI#491) * [AutoDiff] Autodiff 7: Fix header size in AdStackAllocaStmt to match u64 runtime layout (Genesis-Embodied-AI#534) * [AutoDiff] Autodiff 8: Surface LLVM adstack push/pop overflow as a Python exception (Genesis-Embodied-AI#535) * [AutoDiff] Autodiff 9: Guard against LLVM worker-thread stack overflow from large per-task adstack budget (Genesis-Embodied-AI#495) * [AutoDiff] Autodiff 10: Implement adstack for SPIR-V (Genesis-Embodied-AI#490) * [AutoDiff] Autodiff 11: Latent adstack-adjacent fixes (AMDGPU hipFree, flush() keeps ctx_buffers_, always-preallocate) (Genesis-Embodied-AI#536) * [Doc] Add AGENTS.md with instructions for AI agents (Genesis-Embodied-AI#541) * [Bug] Abort kernel execution on assertion failure instead of segfaulting (Genesis-Embodied-AI#419) * [Type] ndarray typing 1: Add eval_str=True to inspect.signature() calls (Genesis-Embodied-AI#411) * [CI] Suppress reportPrivateImportUsage in torch-using files (Genesis-Embodied-AI#552) * [Misc] QD_DUMP_IR dumps to files with the task_id added to the filename (Genesis-Embodied-AI#441) * [Type] ndarray typing 2: Fix NDArray single-arg subscript crash (Genesis-Embodied-AI#412) * [Test] Flush xdist channel before worker exit so test failure reports are visible (Genesis-Embodied-AI#555) * [CI] Reduce test retries on CI from 3 to 1. (Genesis-Embodied-AI#554) * [AutoDiff] Autodiff 12: Heap-backed adstack on LLVM backends (CPU/CUDA/AMDGPU) (Genesis-Embodied-AI#537) * [AutoDiff] Autodiff 13: Heap-backed adstack on SPIR-V backends (Metal, Vulkan) (Genesis-Embodied-AI#493) * [AutoDiff] Autodiff 14: Resolve bounded-inner-loop adstacks without default_ad_stack_size fallback (Genesis-Embodied-AI#539) * [SPIRV] Vulkan SPIR-V correctness: atomic-view aliasing, PSB stride, narrow storage caps, u1 cast, per-init layer recheck (Genesis-Embodied-AI#513) * [Build] Autodiff 15: Replace 2022 MoltenVK pin with LunarG Vulkan SDK fetch and sanitise MoltenVK cap advertisement (Genesis-Embodied-AI#551) * [Test] Suppress stock pytest-timeout to avoid conflict with pytest_hardtle (Genesis-Embodied-AI#557) * [Vulkan] Use SDK validation layer for debugPrintf instead of apt package (Genesis-Embodied-AI#562) * [Test] Fix flaky perf_dispatch tests by increasing work amounts (Genesis-Embodied-AI#559) * [Test] Add --maxfail CLI option to run_tests.py (default 20) (Genesis-Embodied-AI#558) * [CI] Vulkan debug printf fix to address flaky tests (Genesis-Embodied-AI#563) * [Docs] Add a new page to help for first time contributors (Genesis-Embodied-AI#426) Authored-by: v01dxyz <v01dxyz@v01d.xyz> * [AutoDiff] Autodiff 16: Resolve reverse-mode adstack depths per-launch via runtime-evaluated SizeExpr (Genesis-Embodied-AI#543) * Fix: raise error if device memory allocation fails (Genesis-Embodied-AI#451) (Genesis-Embodied-AI#453) Co-authored-by: v01dxyz <v01dxyz@v01d.xyz> Co-authored-by: Hugh Perkins <hughperkins@gmail.com> * [CI] Add CI job to check line wrapping of comments and docs (Genesis-Embodied-AI#564) * [Misc] Add coverage report to PRs, including kernels (Genesis-Embodied-AI#470) * [CI] CI wrap check feeds only diffs to agent (Genesis-Embodied-AI#567) * Skip 'flaky' test on MacOS CI. (Genesis-Embodied-AI#573) * [Test] Fix missing `import sys` in test_fail_device_memory_allocation (Genesis-Embodied-AI#574) * [CI] Fix Vulkan debugPrintf flake with session-scoped warmup (Genesis-Embodied-AI#571) * [AutoDiff] determine_ad_stack_size: replace whole-CFG Bellman-Ford with SCC + DAG DP (Genesis-Embodied-AI#575) * [Test] Fix macOS OOM skip reason to describe actual root cause (Genesis-Embodied-AI#576) * [Lang] whole_kernel_cse: 2.5x compile time speedup on large kernels (Genesis-Embodied-AI#577) * [CI] Add CI check for unnecessarily deleted comments (Genesis-Embodied-AI#570) * [CI] Migrate coverage report to github Check page (Genesis-Embodied-AI#566) * [Lang] Skip IR verifier between passes unless debug=true (Genesis-Embodied-AI#579) * [Lang] Inline AdStack ops on release LLVM codegen: dramatically reduces compile time for adstack-enabled reverse-mode kernels (Genesis-Embodied-AI#584) * [CUDA] Honor offline_cache=False end-to-end so QD_OFFLINE_CACHE=0 actually gives a cold compile (Genesis-Embodied-AI#580) * [Type] Tensor 24 (Genesis-Embodied-AI#561) Co-authored-by: hugh <hugh@slurm-login-0.slurm-login.tenant-slurm.svc.cluster.local> * [Lang] auto_diff host-walk reductions: dramatically faster front-end compile time on adstack-enabled reverse-mode kernels (Genesis-Embodied-AI#587) * [AutoDiff] Speed up reverse-mode kernel launches on GPU backends (Genesis-Embodied-AI#578) * [Vulkan] Move adstack-sizer scratch out of Function-scope memory to fix SPIR-V pipeline build failures (Genesis-Embodied-AI#588) * [AutoDiff] Improve diagnosis of unsupported reverse-mode AD patterns (Genesis-Embodied-AI#590) * [Bug] Fix: promote Ndarray to AnyArray in build_Name for flattened struct fields (Genesis-Embodied-AI#592) * [SPIR-V] Shrink reverse-grad kernel MSL by ~50% (Genesis-Embodied-AI#591) * [CI] Add CI check that PR changes have test coverage (Genesis-Embodied-AI#596) * [Perf] Enable zero-copy in to_torch() and to_numpy() (Genesis-Embodied-AI#450) * Add BufferView: safe sub-range ndarray access for kernels (Genesis-Embodied-AI#585) Co-authored-by: alanray-tech <alanray-tech@users.noreply.github.com> Co-authored-by: Hugh Perkins <hughperkins@gmail.com> * [Doc] Add user-facing fastcache documentation (Genesis-Embodied-AI#597) Co-authored-by: hugh <hugh@slurm-login-0.slurm-login.tenant-slurm.svc.cluster.local> * [Misc] Upgrade to enable v1 dlpack so to_numpy(copy=False) writable (Genesis-Embodied-AI#598) Co-authored-by: root <root@rtx-209-201.slurm-compute.tenant-slurm.svc.cluster.local> * [AutoDiff] Cut reverse-mode adstack memory usage 10x on all backends (Genesis-Embodied-AI#599) * [Misc] Add CI check for feature file factorization (Genesis-Embodied-AI#606) * [Perf] Skip _recursive_set_args for all-Field frozen dataclass structs (Genesis-Embodied-AI#607) Co-authored-by: Cursor <cursoragent@cursor.com> * [AutoDiff] SNode-arm bound-expr capture rejects fold-attack gate indices (Genesis-Embodied-AI#610) * [Misc] Suppress field fastcache warning for qd.Tensor (Genesis-Embodied-AI#615) Co-authored-by: Cursor <cursoragent@cursor.com> * [AutoDiff] Adstack heap: clip reducer count by per-task loop trip count (compile-time and SizeExpr-evaluated) (Genesis-Embodied-AI#611) * [Misc] Forward copy= through qd.Tensor, add copy=None option (Genesis-Embodied-AI#616) Co-authored-by: Cursor <cursoragent@cursor.com> * [Doc] Update README (Genesis-Embodied-AI#617) Co-authored-by: Cursor <cursoragent@cursor.com> * [CI] Fix coverage report showing def lines as uncovered (Genesis-Embodied-AI#623) Co-authored-by: Cursor <cursoragent@cursor.com> * [Perf] Generic launcher: persistent context, JIT-pointer reuse, Metal compute encoder, LLVM-GPU async memory ops (Part 1/2) (Genesis-Embodied-AI#619) * [CI] Encode Python-first testing policy in coverage-check prompt (Genesis-Embodied-AI#622) Co-authored-by: Cursor <cursoragent@cursor.com> * [CI] Add PR Line change report (Genesis-Embodied-AI#624) Co-authored-by: Cursor <cursoragent@cursor.com> * [CI] Disable quadrants pytest plugin during quadrants internal coverage runs (Genesis-Embodied-AI#629) Co-authored-by: Cursor <cursoragent@cursor.com> * [AutoDiff] Adstack load+store eliminations: EliminateRecomputableAdStackPushes pass + leaf extensions (Genesis-Embodied-AI#621) * [CI] Simplify coverage PR comment to a single linked line (Genesis-Embodied-AI#630) * [CUDA] Add AGX Thor, SM_110 (Genesis-Embodied-AI#631) Co-authored-by: Johnny Nunez and Hugh Perkins * [CI] Lines changed report: collapse PR comment to a single linked totals line (Genesis-Embodied-AI#632) * [FEATURE] Support external Metal command queue via qd.init (Genesis-Embodied-AI#618) Co-authored-by: Cursor <cursoragent@cursor.com> * [Perf] Cache adstack-sizer metadata per task across SPIR-V + LLVM-GPU; per-snode / DeviceAllocation invalidation (Part 2/2) (Genesis-Embodied-AI#620) * [AutoDiff] Disable EliminateRecomputableAdStackPushes pending mutated-SNode chain-leaf fix (Genesis-Embodied-AI#633) * [AutoDiff] Adstack chain-clone safety: mutated-SNode leaf reject + load_top consumer-aware guard (Genesis-Embodied-AI#634) * [Docs] Add user-guide page for qd.simt.block.* primitives (Genesis-Embodied-AI#638) * [Docs] Expand qd.simt.subgroup user-guide page to cover every op (Genesis-Embodied-AI#639) * [Perf] Streams 1-4 (Genesis-Embodied-AI#410) * [Docs] Add user-guide page for matrix decompositions and solvers (Genesis-Embodied-AI#643) * [Bug] Revert "[Perf] Streams 1-4 (Genesis-Embodied-AI#410)" (Genesis-Embodied-AI#650) * [Docs] Add user-guide page for atomics and bit operations (Genesis-Embodied-AI#640) * [Docs] Add user-guide page for qd.simt.grid.* primitives (Genesis-Embodied-AI#641) * [AutoDiff] Adstack max-reducer: parallel multi-axis MaxOverRange dispatch (Genesis-Embodied-AI#635) * [AMDGPU] Fix amdgpu parallel rand init (Genesis-Embodied-AI#658) * [Perf] Adstack: skip max-reducer recognizer on CPU + lift host-eval cap (Genesis-Embodied-AI#655) * [Perf] Re-land Streams 1-4 with bug fixes (Genesis-Embodied-AI#653) * [AMDGPU] Apply device_memory_GB=0.3 cap to AMDGPU tests (Genesis-Embodied-AI#659) * [Perf] Per-launch host sync: drop wait_idle on SPIR-V, pin stream and drop stream_synchronize on CUDA/AMDGPU (Genesis-Embodied-AI#654) * [AMDGPU] Unload hipModule_t in JITModuleAMDGPU destructor (Genesis-Embodied-AI#660) * [AMDGPU] Trim default mempool on qd.reset() (Genesis-Embodied-AI#669) * [AMDGPU] Hoist rand-state buffer to process lifetime (Genesis-Embodied-AI#668) * [Streams] Use events for streams serialization on AMDGPU and CUDA (Genesis-Embodied-AI#667) * [Perf] Adstack max-reducer: launch cache + zero-copy result map; content-stable registry_id (Genesis-Embodied-AI#671) * [SPIR-V] dispatch_max_reducers: register each task with the real kernel name (Genesis-Embodied-AI#675) * [AutoDiff] Debug-mode field/grad/dual: dtype, layout, and access-time invariants (Genesis-Embodied-AI#677) * [Docs] Add user-guide page for qd.algorithms.* device-wide algorithms (Genesis-Embodied-AI#642) Co-authored-by: alanray-tech <alan.ray@genesis-ai.company> * [Docs] Doc for existing atomics: switch support table to per-backend columns (Genesis-Embodied-AI#657) Co-authored-by: alanray-tech <alan.ray@genesis-ai.company> * [GPU] Cross gpu atomics (Genesis-Embodied-AI#666) Co-authored-by: alanray-tech <alan.ray@genesis-ai.company> * [GPU] Make block operations portable cross-gpu (Genesis-Embodied-AI#664) * [Perf] CPU LLVM adstack-cache: skip per-launch bump-writes + ndarray_shapes capture on forward-only handles (Genesis-Embodied-AI#685) * [GPU] Cross-GPU for grid ops (Genesis-Embodied-AI#670) * [Math] Make bitop operations portable cross-gpu (Genesis-Embodied-AI#662) * [AMDGPU] Always use wave64, on both RDNA and CDNA (Genesis-Embodied-AI#687) * [AMDGPU] Use syncscope("agent") for atomix xor to avoid CAS livelock (Genesis-Embodied-AI#672) * [GPU] New bit ops for QIPC (Genesis-Embodied-AI#679) * [GPU] Subgroup ops cross-gpu (Genesis-Embodied-AI#665) * [Graph] Rename CUDA Graph to Graph in docs (Genesis-Embodied-AI#691) * [SPIR-V] Fix FIFO-queue ordering when sharing command queue. (Genesis-Embodied-AI#694) * [Atomics] New QIPC ops for atomics (Genesis-Embodied-AI#690) * Pass dataclass sub-structs into qd.func (Genesis-Embodied-AI#698) * [AMDGPU] HIP graph runtime support for @qd.kernel(graph=True) (Genesis-Embodied-AI#692) * [CI] Add per-file timing report to Mac Metal test job (Genesis-Embodied-AI#695) Co-authored-by: Cursor <cursoragent@cursor.com> * [CI] Enable kernel disk cache during tests (Genesis-Embodied-AI#696) * [Math] New QIPC ops for single-threaded linalg (Genesis-Embodied-AI#683) * [BREAKING][GPU] New QIPC ops for subgroups (Genesis-Embodied-AI#676) * [GPU] New QIPC ops for block (Genesis-Embodied-AI#684) * [GPU] New device-level ops for QIPC (Genesis-Embodied-AI#693) * [algorithms] PrefixSumExecutor: drop unused GRID_SZ local (Genesis-Embodied-AI#701) * [block] sync(): fix unsupported-arch error message (Genesis-Embodied-AI#700) * [volatile_load] add qd.volatile_load primitive (closes Genesis-Embodied-AI#648) (Genesis-Embodied-AI#702) * [AutoDiff] Reject recycled identity_key in AdStackCache::register_adstack_sizing_info (Genesis-Embodied-AI#708) * [Vulkan] Declare GroupNonUniform SPIR-V caps and enable shaderSubgroupExtendedTypes (Genesis-Embodied-AI#707) * Fix duplicate HIP graph driver-function declarations after v1.0.0 merge The amd-integration fork had cherry-picked the HIP graph driver functions (graph_create / graph_destroy / graph_add_kernel_node / graph_instantiate / graph_exec_destroy / graph_launch), and upstream v1.0.0 added the same set. The per-file 3-way merge appended both copies into amdgpu_driver_functions.inc.h, producing redeclaration errors that broke the AMDGPU RHI/runtime compile. Drop the upstream duplicate block; the signatures are identical to the fork's existing declarations. Co-authored-by: Cursor <cursoragent@cursor.com> * Fix AMDGPU launcher coherence and num_instructions visibility after v1.0.0 merge - kernel_launcher.cpp: the 3-way merge spliced upstream v1.0.0's launch_llvm_kernel rewrite (ephemeral arg/context buffers, explicit-stream path, AmdgpuDefaultStream PinGuard) onto the AMD fork's kernarg-by-value + persistent-scratch design, leaving references to undefined `ephemeral_context_ptr`. Restore the fork's coherent launch_llvm_kernel verbatim; it calls the (already merged) enhanced launch_offloaded_tasks, which keeps the max-reducer dispatch and stream-parallel groups adapted onto the AMD launch path. - llvm_context.h: both the fork and upstream added `num_instructions`; the merge kept upstream's private placement, but the AMDGPU codegen force-inline heuristic calls it statically from outside the class. Move it back to the public section. Co-authored-by: Cursor <cursoragent@cursor.com> * Restore async result D2H and hoist kernarg vectors in AMDGPU launcher The v1.0.0 merge resolution regressed two amd-integration baseline optimizations in launch_llvm_kernel / launch_offloaded_tasks: - The per-launch result-buffer copy was a blocking memcpy_device_to_host, forcing a host stall on every value-returning launch and serializing the GPU pipeline. Restore the async D2H (the caller synchronizes lazily when it needs the value); external-array transfers still stream_synchronize once before reading back. - launch_task constructed the kernarg std::vectors from initializer lists ({kernarg_payload} / {kernarg_size}) on every dispatch (heap alloc + free per launch). Hoist arg_ptrs/arg_sizes out of the per-task launch and reuse. Co-authored-by: Cursor <cursoragent@cursor.com> * amdgpu: default to LDS permlane64 emulation; drop host-x86 barrier asm on retarget Two AMDGPU JIT-compile crashes surfaced after the v1.0.0 merge pulled in the QIPC subgroup ops (Genesis-Embodied-AI#676), which made the rigid constraint solver's wave-cooperative reductions route through `amdgpu_cross_half_shuffle_i32`. Both manifested as a SIGSEGV inside `llvm::SIInstrInfo::getInstSizeInBytes` during `JITSessionAMDGPU::compile_module_to_hsaco` (i.e. at first kernel launch), and reproduce on gfx942 / MI300X. Baseline 0.4.6 never emitted these constructs, which is why it was unaffected. 1. Native `llvm.amdgcn.permlane64` lowering crashes the bundled LLVM 22.1.0 AMDGPU backend. Default `amdgpu_permlane64` to the existing LDS-roundtrip software emulation on every target (it produces identical results). Add `QD_AMDGPU_USE_NATIVE_PERMLANE64=1` to opt back into the native instruction once the backend bug is fixed; the old `QD_AMDGPU_FORCE_PERMLANE64_FALLBACK` is now the default and still honored. This is the actual crash fix. 2. The runtime module is compiled by the host x86_64 clang and only retargeted to amdgcn here, so `amdgpu_cross_half_shuffle_i32`'s `__asm__ volatile("" : "+v"(byte))` optimization barrier carries x86 flag clobbers (`~{dirflag},~{fpsr},~{flags}`) that are meaningless on AMDGPU. The IR verifies but the empty-body INLINEASM is invalid on the amdgcn target. Neutralize empty-body barrier asm during retarget (forward the tied value, then erase) so no stale host asm reaches codegen. On the wave64 targets we ship `ds_bpermute` already addresses the full wave, so the hint is a no-op. Co-authored-by: Cursor <cursoragent@cursor.com> * style: apply clang-format (v19.1.7) to AMDGPU fn_attrs and launcher sources CI pre-commit's clang-format hook reformatted these files (long declarations/lambda signatures collapsed onto single lines per the repo's clang-format config). Apply the same formatting so the hook passes. No functional changes. Co-authored-by: Cursor <cursoragent@cursor.com> * fix(amdgpu): use CreateNeg for branchless i32 sgn instead of CreateSub(0, input) clang-tidy (modernize-use-nullptr, -warnings-as-errors) flagged `builder->CreateSub(0, input)` in the i32 sgn path: the literal `0` binds to the `llvm::Value*` LHS parameter as a null pointer, not an integer zero. Replace with `builder->CreateNeg(input)`, which emits `0 - input` with a proper zero constant -- identical intended semantics, and clang-tidy clean. Co-authored-by: Cursor <cursoragent@cursor.com> --------- Co-authored-by: Robert Dazi <14996868+v01dXYZ@users.noreply.github.com> Co-authored-by: v01dxyz <v01dxyz@v01d.xyz> Co-authored-by: Hugh Perkins <hughperkins@gmail.com> Co-authored-by: Alexis DUBURCQ <alexis.duburcq@gmail.com> Co-authored-by: hugh <hugh@slurm-login-0.slurm-login.tenant-slurm.svc.cluster.local> Co-authored-by: alanray-tech <alan.ray@genesis-ai.company> Co-authored-by: alanray-tech <alanray-tech@users.noreply.github.com> Co-authored-by: root <root@rtx-209-201.slurm-compute.tenant-slurm.svc.cluster.local> Co-authored-by: Cursor <cursoragent@cursor.com> Co-authored-by: Johnny <johnnynuca14@gmail.com>
The NVIDIA Vulkan driver has a bug where repeated
vkDestroyInstance/vkCreateInstance cycles corrupt
SubgroupLocalInvocationId after ~11 cycles. The corruption is permanent for the process lifetime and cannot be recovered from, even with VK_EXT_subgroup_size_control + requiredSubgroupSize=32.
Fix: keep the VkInstance alive in VulkanLoader (process-lifetime singleton) and reuse it across qd.init/qd.reset cycles. Only the VkDevice is destroyed and recreated.
Verified: 30 consecutive Vulkan init/reset cycles all produce correct SubgroupLocalInvocationId = [0..15].
Reproducer: standalone C program (tmp/vk_subgroup_repro.c) confirms the bug is in the NVIDIA driver, not in Quadrants code.
Issue: #
Brief Summary
copilot:summary
Walkthrough
copilot:walkthrough