From 88513cfd9c5896e8760a7ab2d51e2febd5055b8b Mon Sep 17 00:00:00 2001 From: Azad Free Date: Fri, 19 Apr 2024 12:50:54 -0700 Subject: [PATCH] removing base_module for glow/fb/torch_glow/py/NNPI:torch_glow Summary: X-link: https://github.com/pytorch/glow/pull/6173 This is an attempt to reduce the number of buck targets with unnecessary base_module. Differential Revision: D56368043 --- inference/compute/pt/pytorch_linear.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference/compute/pt/pytorch_linear.py b/inference/compute/pt/pytorch_linear.py index 9c946be9..7fea918d 100644 --- a/inference/compute/pt/pytorch_linear.py +++ b/inference/compute/pt/pytorch_linear.py @@ -33,7 +33,7 @@ def forward(self, x): def infer_nnpi(model, device, data_type, input_size, output_size, batch_size, args): - import torch_glow + import glow.fb.torch_glow.py.NNPI.torch_glow as torch_glow # Detailed structure for spec can be found at https://fburl.com/diffusion/79q4efud # Create compilation spec