mirror of
https://github.com/ROCm/ROCm.git
synced 2026-04-05 03:01:17 -04:00
[FRONTEND] Expose the register usage and spill information thru CompiledKernel (#1296)
This commit is contained in:
@@ -1725,6 +1725,8 @@ class CompiledKernel:
|
||||
if self.shared > max_shared:
|
||||
raise OutOfResources(self.shared, max_shared, "shared memory")
|
||||
mod, func, n_regs, n_spills = cuda_utils.load_binary(self.metadata["name"], self.asm["cubin"], self.shared, device)
|
||||
self.n_spills = n_spills
|
||||
self.n_regs = n_regs
|
||||
self.cu_module = mod
|
||||
self.cu_function = func
|
||||
|
||||
|
||||
Reference in New Issue
Block a user