Add try/except OSError guard for torchrec/quant/embedding_modules.py torch.ops.load_library call.

rkindi · facebook-github-bot · commit 424994a79506 · 2021-12-02T17:55:27.000-08:00
Summary: Everywhere else (torchrec/sparse/jagged_tensor.py, torchrec/distributed/comm_ops.py, torchrec/distributed/dist_data.py) we have a try except guard.

Reviewed By: colin2328

Differential Revision: D32814873

fbshipit-source-id: 467a0c5ed13b1ae059f8c42aa1e33549da785ef3
diff --git a/torchrec/quant/embedding_modules.py b/torchrec/quant/embedding_modules.py
@@ -28,7 +28,10 @@
     KeyedTensor,
 )
 
-torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops")
+try:
+    torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops")
+except OSError:
+    pass
 
 
 class EmbeddingBagCollection(EmbeddingBagCollectionInterface):

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,10 @@`
`28`	`28`	`KeyedTensor,`
`29`	`29`	`)`
`30`	`30`
`31`		`-torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops")`
	`31`	`+try:`
	`32`	`+ torch.ops.load_library("//deeplearning/fbgemm/fbgemm_gpu:sparse_ops")`
	`33`	`+except OSError:`
	`34`	`+ pass`
`32`	`35`
`33`	`36`
`34`	`37`	`class EmbeddingBagCollection(EmbeddingBagCollectionInterface):`