From ec4f57166b5e03758ea901be701a1042516b48f8 Mon Sep 17 00:00:00 2001 From: whf <1660398197@qq.com> Date: Fri, 13 Jun 2025 10:14:32 +0800 Subject: [PATCH] =?UTF-8?q?=E8=B5=84=E6=96=99=E6=B2=A1=E6=9C=89=E5=86=99ta?= =?UTF-8?q?ble=20wise=E7=9A=84=E6=94=AF=E6=8C=81=E5=92=8Cdemo=E9=87=8C?= =?UTF-8?q?=E9=9D=A2=E4=BD=BF=E7=94=A8=E7=9A=84table=20wise?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- torchrec/example/little_demo/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrec/example/little_demo/main.py b/torchrec/example/little_demo/main.py index 5fac659f..edbf4003 100644 --- a/torchrec/example/little_demo/main.py +++ b/torchrec/example/little_demo/main.py @@ -91,7 +91,7 @@ def invoke_main(): hybrid_sharder = get_default_hybrid_sharders(host_env=host_env) constrans = { table_name: ParameterConstraints( - sharding_types=["table_wise"], compute_kernels=["fused"] + sharding_types=["row_wise"], compute_kernels=["fused"] ) for table_name in TABLE_NAMES } -- Gitee