Skip to content

Commit

Permalink
change to single thread to compare and debug
Browse files Browse the repository at this point in the history
  • Loading branch information
BolinSNLHM committed Mar 28, 2024
1 parent e5f9b0f commit 65fc51b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/hidet/graph/ops/matmul/matmul_f32_x86.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def implement_cpu(self, working_dir: str) -> Union[IRModule, List[IRModule]]:
return tune.extract_ir_modules(self.schedule_matmulf32_x86)

@tune.space(1, MC=[2016], NC=[256, 384, 512], KC=[384, 512, 560], ways=[(1, 4, 2, 1)])
def schedule_matmulf32_x86(self, MC=2016, NC=384, KC=560, ways=(1, 4, 2, 1)) -> IRModule:
def schedule_matmulf32_x86(self, MC=2016, NC=384, KC=560, ways=(1, 1, 1, 1)) -> IRModule:
import hidet
from hidet.ir.type import tensor_type
from hidet.lang import tensor, grid, as_tensor_pointer
Expand Down

0 comments on commit 65fc51b

Please sign in to comment.