another torch deploy protection :( [guard import in torchrec train pipeline] (#2512)

Summary:
Pull Request resolved: #2512

this import doesnt work in torch dceploy for whatever reason (we're getting rid of it all in 2 months so just bare with this for now)

Differential Revision: D64830919

fbshipit-source-id: 8271244fef2697f85e361b4fd1b3937f549f0994

Author: s4ayub

torchrec/distributed/train_pipeline/train_pipelines.py

@@ -74,7 +74,8 @@
 except ImportError:
     logger.warning("torchrec_use_sync_collectives is not available")
 
-torch.ops.import_module("fbgemm_gpu.sparse_ops")
+if not torch._running_with_deploy():
+    torch.ops.import_module("fbgemm_gpu.sparse_ops")
 
 
 class ModelDetachedException(Exception):