few type cleanups [pr] (#8347)

This commit is contained in:
chenyu
2024-12-20 01:56:01 -05:00
committed by GitHub
parent 82833f1b3c
commit e63c7818dc
6 changed files with 12 additions and 13 deletions

View File

@@ -146,7 +146,7 @@ class AMDComputeQueue(HWQueue):
self.indirect_cmd = [amd_gpu.PACKET3(amd_gpu.PACKET3_INDIRECT_BUFFER, 2), *data64_le(self.hw_page.va_addr),
len(self._q) | amd_gpu.INDIRECT_BUFFER_VALID]
self._q = hw_view # type: ignore
self._q = hw_view
return self
def _submit(self, dev:AMDDevice):

View File

@@ -5,7 +5,7 @@
# it should be a secure (example: no use of pickle) boundary. HTTP is used for RPC
from __future__ import annotations
from typing import Optional, Any, DefaultDict
from typing import Optional, Any
from collections import defaultdict
from dataclasses import dataclass, field
import multiprocessing, functools, http.client, hashlib, json, time, os, binascii, struct, ast, contextlib
@@ -81,7 +81,7 @@ class CloudSession:
class CloudHandler(BaseHTTPRequestHandler):
protocol_version = 'HTTP/1.1'
device: str
sessions: DefaultDict[str, CloudSession] = defaultdict(CloudSession)
sessions: defaultdict[str, CloudSession] = defaultdict(CloudSession)
def setup(self):
super().setup()

View File

@@ -111,7 +111,7 @@ class NVCommandQueue(HWQueue[NVSignal, 'NVDevice', 'NVProgram', 'NVArgsState']):
for i, value in enumerate(self._q): hw_view[i] = value
# From now on, the queue is on the device for faster submission.
self._q = hw_view # type: ignore
self._q = hw_view
def _submit_to_gpfifo(self, dev:NVDevice, gpfifo:GPFifo):
if dev == self.binded_device: cmdq_addr = self.hw_page.va_addr

View File

@@ -97,7 +97,7 @@ class QCOMComputeQueue(HWQueue):
self.hw_page = dev.allocator.alloc(len(self._q) * 4, BufferSpec(cpu_access=True, nolru=True))
self.submit_req, self.obj = self._build_gpu_command(self.binded_device, self.hw_page.va_addr)
# From now on, the queue is on the device for faster submission.
self._q = to_mv(self.obj.gpuaddr, len(self._q) * 4).cast("I") # type: ignore
self._q = to_mv(self.obj.gpuaddr, len(self._q) * 4).cast("I")
def _submit(self, dev:QCOMDevice):
if self.binded_device == dev: submit_req = self.submit_req