Class FloatingType

cutlass.py/dtype.py:96–156

@dataclass
class FloatingType(DType):
    sign_bits: int = 0
    exp_bits: int = 0
    mantissa_bits: int = 0

    @property
    def bits(self):
        return self.sign_bits + self.exp_bits + self.mantissa_bits

    def _is(self, sign: int, exp: int, mantissa: int):
        return (
            self.sign_bits == sign
            and self.exp_bits == exp
            and self.mantissa_bits == mantissa
        )

    def is_e3m4_t(self):
        return self._is(1, 3, 4)

    def is_e4m3_t(self):
        return self._is(1, 4, 3)

    def is_fp8_t(self):
        return self.is_e3m4_t() or self.is_e4m3_t()

    def is_half_t(self):
        return self._is(1, 5, 10)

    def is_float_t(self):
        return self._is(1, 8, 23)

    def is_double_t(self):
        return self._is(1, 11, 52)

    @classmethod
    def e3m4_t(cls):
        return cls(1, 3, 4)

    @classmethod
    def e4m3_t(cls):
        return cls(1, 4, 3)

    @classmethod
    def half_t(cls):
        return cls(1, 5, 10)

    @classmethod
    def float_t(cls):
        return cls(1, 8, 23)

    @classmethod
    def double_t(cls):
        return cls(1, 11, 52)

    def is_same(self, another_type: "FloatingType"):
        return (
            isinstance(another_type, FloatingType)
            and (self.sign_bits == another_type.sign_bits)