fbgemm op ci test fail

Differential Revision: D36133994 fbshipit-source-id: 9b1b155eeaa6345c1cdd388dc91cbc7babe44265
facebookresearch · May 4, 2022 · 2aef7c6 · 2aef7c6
1 parent 17799b0
commit 2aef7c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/reagent/test/preprocessing/test_transforms.py b/reagent/test/preprocessing/test_transforms.py
@@ -9,6 +9,7 @@
 import numpy as np
 import reagent.core.types as rlt
 import torch
+import fbgemm_gpu
 from reagent.preprocessing import transforms
 from reagent.preprocessing.types import InputColumn
 

diff --git a/setup.cfg b/setup.cfg
@@ -17,6 +17,7 @@ install_requires =
   # issue: https://github.com/openai/spinningup/issues/178
   cloudpickle~=1.2.0
   iopath
+  fbgemm-gpu-nightly
   numpy>=1.17.2
   pandas>=1.0.3
   # https://github.com/samuelcolvin/pydantic/issues/2042