Import sparse_ops

Differential Revision: D57648027
Microve · May 21, 2024 · fde9f64 · fde9f64
1 parent f9beba1
commit fde9f64
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchrec/sparse/jagged_tensor.py b/torchrec/sparse/jagged_tensor.py
@@ -13,6 +13,8 @@
 
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
+import fbgemm_gpu.sparse_ops  # noqa: F401, E402
+
 import torch
 from torch.autograd.profiler import record_function
 from torch.fx._pytree import register_pytree_flatten_spec, TreeSpec