github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix OSX image pitch (#14193)

Browse Source
This commit is contained in:
Christopher Milan
2026-01-16 16:07:33 -08:00
committed by GitHub
parent 3960e2758c
commit eafcd44d95
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/dtype.py
View File

@@ -2,7 +2,7 @@ from __future__ import annotations
from typing import Final, ClassVar, Callable, Literal
import math, struct, ctypes, functools
from dataclasses import dataclass, fields
from tinygrad.helpers import getenv, prod, round_up, next_power2
from tinygrad.helpers import getenv, prod, round_up, next_power2, OSX
from enum import Enum, auto
class InvalidType:
@@ -105,6 +105,7 @@ class ImageDType(PtrDType):
def pitch(self):
if self._pitch != -1: return self._pitch
imgw, imgh, itemsize_log = self.shape[1], self.shape[0], int(math.log2(self.itemsize))
if OSX: return round_up(imgw, 256) * 4 * self.itemsize
pitchalign = max(6, 11 - int(math.log2(imgh))) if imgh > 1 else 6
align_up = max(1, (8 // itemsize_log + 1) - imgh // 32) if pitchalign == 6 else (2 ** (pitchalign - itemsize_log - 2))