Add support for Rowwise and Columnwise sharding (#1621)

Summary:
Pull Request resolved: #1621

Add in Rowwise and Columnwise sharding to inference module benchmarking suite by default.

Reviewed By: bigning

Differential Revision: D52641307

fbshipit-source-id: f396bcfb9ac71fe65906a723184efde547c80235

torchrec/distributed/benchmark/benchmark_inference.py

@@ -439,11 +439,10 @@ def main() -> None:
         # Place all outputs under the datetime folder
         os.mkdir(output_dir)
 
-    # TODO: ROW_WISE and COLUMN_WISE are not supported yet
     BENCH_SHARDING_TYPES = [
         ShardingType.TABLE_WISE,
-        # ShardingType.ROW_WISE,
-        # ShardingType.COLUMN_WISE,
+        ShardingType.ROW_WISE,
+        ShardingType.COLUMN_WISE,
     ]
 
     table_sizes = [