(self)
| 1967 | { |
| 1968 | "logical_mesh": HybridMeshShape(ici_mesh_shape=(1, -1), dcn_mesh_shape=(2, 1)), |
| 1969 | "expected": (2, 16), |
| 1970 | }, |
| 1971 | # Test a case where we infer -1 in DCN mesh. |
| 1972 | { |
| 1973 | "logical_mesh": HybridMeshShape(ici_mesh_shape=(1, 16), dcn_mesh_shape=(-1, 1)), |
| 1974 | "expected": (2, 16), |
| 1975 | }, |
| 1976 | # Test a case where we infer -1 in both ICI and DCN mesh. |
| 1977 | { |
| 1978 | "logical_mesh": HybridMeshShape(ici_mesh_shape=(1, -1), dcn_mesh_shape=(-1, 1)), |
| 1979 | "expected": (2, 16), |
nothing calls this directly
no test coverage detected