fbgemm op ci test fail (#638)

Summary: Pull Request resolved: #638 Differential Revision: D36133994 fbshipit-source-id: d76a42b84f3eab4196a5d4f8210f3f37c5edf55e
facebookresearch · May 4, 2022 · 31fc8d9 · 31fc8d9
1 parent 8889cf2
commit 31fc8d9
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