| 1056 | ctx_save_restore_size=ctx_save_restore_size, ctl_stack_size=ctl_stack_size, idx=idx)) |
| 1057 | |
| 1058 | def sdma_queue(self, idx:int): |
| 1059 | if getenv("AMD_DISABLE_SDMA"): return None |
| 1060 | if idx in self.sdma_queues: return self.sdma_queues[idx] |
| 1061 | with contextlib.suppress(OSError): |
| 1062 | self.sdma_queues[idx] = self.create_queue(kfd.KFD_IOC_QUEUE_TYPE_SDMA, 0x200 if self.is_usb() else (16 << 20), idx=idx) |
| 1063 | return self.sdma_queues.get(idx, None) |
| 1064 | |
| 1065 | def _ensure_has_local_memory(self, private_segment_size): |
| 1066 | if self.max_private_segment_size >= private_segment_size: return |