Test cpu ops with non-default memory layouts (#39)

Summary:
Pull Request resolved: #39

imported-using-ghimport

Test Plan:
Imported from OSS

Rollback Plan:

Reviewed By: digantdesai

Differential Revision: D80468302

Pulled By: manuelcandales

fbshipit-source-id: 75e018e5e0092de8988c253d7438f3fc9692f721

Author: manuelcandales

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()