diff --git a/test/specdb/test_specdb_cpu.py b/test/specdb/test_specdb_cpu.py index bfeb7eb..b67e1a4 100644 --- a/test/specdb/test_specdb_cpu.py +++ b/test/specdb/test_specdb_cpu.py @@ -35,6 +35,37 @@ def test_all_ops_cpu_half(self): config = TensorConfig(device="cpu", half_precision=True) self._run_all_ops(config=config, skip_ops=skip_ops) + def test_all_ops_cpu_transposed(self): + skip_ops = self.SKIP_OPS.copy() + # Expected X.is_contiguous(memory_format) to be true, but got false. + skip_ops += ["native_group_norm.default"] + # _pdist_forward requires contiguous input + skip_ops += ["_pdist_forward.default"] + config = TensorConfig(device="cpu", transposed=True) + self._run_all_ops(config=config, skip_ops=skip_ops) + + def test_all_ops_cpu_permuted(self): + skip_ops = self.SKIP_OPS.copy() + # Expected X.is_contiguous(memory_format) to be true, but got false. + skip_ops += ["native_group_norm.default"] + # _pdist_forward requires contiguous input + skip_ops += ["_pdist_forward.default"] + # Unsupported memory format. Supports only ChannelsLast3d, Contiguous + skip_ops += ["max_pool3d_with_indices.default"] + # Unsupported memory format. Supports only ChannelsLast, Contiguous + skip_ops += ["pixel_shuffle.default"] + config = TensorConfig(device="cpu", permuted=True) + self._run_all_ops(config=config, skip_ops=skip_ops) + + def test_all_ops_cpu_strided(self): + skip_ops = self.SKIP_OPS.copy() + # Expected X.is_contiguous(memory_format) to be true, but got false. + skip_ops += ["native_group_norm.default"] + # _pdist_forward requires contiguous input + skip_ops += ["_pdist_forward.default"] + config = TensorConfig(device="cpu", strided=True) + self._run_all_ops(config=config, skip_ops=skip_ops) + if __name__ == "__main__": unittest.main()