()
| 25 | class Cast(Base): |
| 26 | @staticmethod |
| 27 | def export() -> None: |
| 28 | test_cases = [ |
| 29 | ("FLOAT", "FLOAT16"), |
| 30 | ("FLOAT", "DOUBLE"), |
| 31 | ("FLOAT16", "FLOAT"), |
| 32 | ("FLOAT16", "DOUBLE"), |
| 33 | ("DOUBLE", "FLOAT"), |
| 34 | ("DOUBLE", "FLOAT16"), |
| 35 | ("FLOAT", "BFLOAT16"), |
| 36 | ("BFLOAT16", "FLOAT"), |
| 37 | ("FLOAT", "FLOAT8E4M3FN"), |
| 38 | ("FLOAT16", "FLOAT8E4M3FN"), |
| 39 | ("FLOAT", "FLOAT8E4M3FNUZ"), |
| 40 | ("FLOAT16", "FLOAT8E4M3FNUZ"), |
| 41 | ("FLOAT8E4M3FN", "FLOAT"), |
| 42 | ("FLOAT8E4M3FN", "FLOAT16"), |
| 43 | ("FLOAT8E4M3FNUZ", "FLOAT"), |
| 44 | ("FLOAT8E4M3FNUZ", "FLOAT16"), |
| 45 | ("FLOAT", "FLOAT8E5M2"), |
| 46 | ("FLOAT16", "FLOAT8E5M2"), |
| 47 | ("FLOAT", "FLOAT8E5M2FNUZ"), |
| 48 | ("FLOAT16", "FLOAT8E5M2FNUZ"), |
| 49 | ("FLOAT8E5M2", "FLOAT"), |
| 50 | ("FLOAT8E5M2", "FLOAT16"), |
| 51 | ("FLOAT8E5M2FNUZ", "FLOAT"), |
| 52 | ("FLOAT8E5M2FNUZ", "FLOAT16"), |
| 53 | ("FLOAT", "UINT4"), |
| 54 | ("FLOAT16", "UINT4"), |
| 55 | ("FLOAT", "INT4"), |
| 56 | ("FLOAT16", "INT4"), |
| 57 | ("UINT4", "FLOAT"), |
| 58 | ("UINT4", "FLOAT16"), |
| 59 | ("UINT4", "UINT8"), |
| 60 | ("INT4", "FLOAT"), |
| 61 | ("INT4", "FLOAT16"), |
| 62 | ("INT4", "INT8"), |
| 63 | ("FLOAT4E2M1", "FLOAT"), |
| 64 | ("FLOAT4E2M1", "FLOAT16"), |
| 65 | ("FLOAT", "FLOAT4E2M1"), |
| 66 | ("FLOAT16", "FLOAT4E2M1"), |
| 67 | ("FLOAT", "UINT2"), |
| 68 | ("FLOAT16", "UINT2"), |
| 69 | ("FLOAT", "INT2"), |
| 70 | ("FLOAT16", "INT2"), |
| 71 | ("UINT2", "FLOAT"), |
| 72 | ("UINT2", "FLOAT16"), |
| 73 | ("UINT2", "UINT8"), |
| 74 | ("INT2", "FLOAT"), |
| 75 | ("INT2", "FLOAT16"), |
| 76 | ("INT2", "INT8"), |
| 77 | ] |
| 78 | |
| 79 | for from_type, to_type in test_cases: |
| 80 | if from_type == to_type: |
| 81 | # Skip cases where from_type and to_type are the same |
| 82 | continue |
| 83 | from_dtype = getattr(TensorProto, from_type) |
| 84 | to_dtype = getattr(TensorProto, to_type) |
nothing calls this directly
no test coverage detected