From 98adbc450c126378069f617cf467a0a2014952ee Mon Sep 17 00:00:00 2001 From: Linjian Ma Date: Tue, 10 Dec 2024 14:51:14 -0800 Subject: [PATCH] remove oss Summary: X-link: https://github.com/facebookresearch/generative-recommenders/pull/163 Still keeps hammer/oss/generative_recommenders/ops/cpp for predictor bc. App layer overlay update in D67013789 Reviewed By: xing-liu Differential Revision: D66943707 --- tritonbench/operators/gemm/operator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tritonbench/operators/gemm/operator.py b/tritonbench/operators/gemm/operator.py index d594c6e..2878249 100644 --- a/tritonbench/operators/gemm/operator.py +++ b/tritonbench/operators/gemm/operator.py @@ -41,7 +41,7 @@ ) if IS_FBCODE: - import hammer.oss.generative_recommenders.ops.triton.triton_addmm as hstu_triton_addmm + import generative_recommenders.ops.triton.triton_addmm as hstu_triton_addmm # without this set we can only pick a single config for AMD, Nvidia has 8 # with this set AMD will pick from 256 different configs (not the actual full