Skip to content

Commit

Permalink
removing base_module for glow/fb/torch_glow/py/NNPI:torch_glow
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/glow#6173

This is an attempt to reduce the number of buck targets with unnecessary base_module.

Differential Revision: D56368043
  • Loading branch information
azad-meta authored and facebook-github-bot committed Apr 19, 2024
1 parent 7868e09 commit 88513cf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion inference/compute/pt/pytorch_linear.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 88513cf

Please # to comment.