Method test_fp8_tensor_conversion

exir/tests/test_tensor.py:94–100

(self)

        # (3*4*5, 1, 5*3, 3))

    def test_fp8_tensor_conversion(self) -> None:
        for dtype in (torch.float8_e5m2, torch.float8_e4m3fn):
            normal_tensor = torch.randn(2, 2, 3, dtype=torch.float32).to(dtype)
            flatbuffer_tensor = make_tensor_value(
                1, 0, TensorSpec.from_tensor(normal_tensor)
            )
            self.compare_tensors(normal_tensor, flatbuffer_tensor)

    def test_allocation_info_succeeds(self) -> None:
        test_cases = (