mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-02-16 01:26:29 -05:00
few type cleanups [pr] (#8347)
This commit is contained in:
@@ -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):
|
||||
|
||||
@@ -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()
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user