File tree Expand file tree Collapse file tree 2 files changed +4
-6
lines changed
python/tvm/contrib/cutlass Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Original file line number Diff line number Diff line change 2222from .conv2d_profiler import Conv2dProfilerEmitter
2323from .gen_tensor_op import (
2424 ProfilerEngine ,
25- generate_sm75_tensor_op_1688 ,
26- generate_sm80_tensor_op_16816 ,
2725 GENERATOR_FUNC_TABLE ,
2826)
2927from .library import (
@@ -187,8 +185,10 @@ def profile(
187185 if profile_all :
188186 self .engine .compile_all (ops , use_multiprocessing )
189187
190- args = ("--n=%d --h=%d --w=%d --c=%d --k=%d --r=%d --s=%d --pad_h=%d --pad_w=%d "
191- "--stride_h=%d --stride_w=%d --dilation_h=%d --dilation_w=%d" ) % workload
188+ args = (
189+ "--n=%d --h=%d --w=%d --c=%d --k=%d --r=%d --s=%d --pad_h=%d --pad_w=%d "
190+ "--stride_h=%d --stride_w=%d --dilation_h=%d --dilation_w=%d"
191+ ) % workload
192192
193193 for op in ops :
194194 out = self .engine .evaluate (op , args .split (" " ))
Original file line number Diff line number Diff line change 2222from .gemm_profiler import GemmProfilerEmitter
2323from .gen_tensor_op import (
2424 ProfilerEngine ,
25- generate_sm75_tensor_op_1688 ,
26- generate_sm80_tensor_op_16816 ,
2725 GENERATOR_FUNC_TABLE ,
2826)
2927from .library import (
You can’t perform that action at this time.
0 commit comments