Commit 0d1eab57 authored by Taoyu Zhu's avatar Taoyu Zhu Committed by LeiWang1999
Browse files

Fix deepgemm exmaple (#513)



* fix deepgemm example

* fix deepgemm example

* make format

* Update example_deepgemm_fp8_2xAcc.py

---------
Co-authored-by: default avatarLei Wang <34334180+LeiWang1999@users.noreply.github.com>
parent de028927
...@@ -6,7 +6,7 @@ import tilelang as TL ...@@ -6,7 +6,7 @@ import tilelang as TL
import tilelang.language as T import tilelang.language as T
from tilelang.utils.tensor import map_torch_type from tilelang.utils.tensor import map_torch_type
tilelang.testing.set_random_seed(0) tilelang.testing.set_random_seed(42)
def tl_gemm( def tl_gemm(
...@@ -143,8 +143,8 @@ def calc_diff(x, y): ...@@ -143,8 +143,8 @@ def calc_diff(x, y):
return 1 - sim return 1 - sim
def assert_tl_gemm_correctness(M, N, K, in_dtype, out_dtype, accum_dtype): def assert_tl_gemm_correctness(M, N, K, block_N, in_dtype, out_dtype, accum_dtype):
gemm = tl_gemm(M, N, K, in_dtype, out_dtype, accum_dtype) gemm = tl_gemm(M, N, K, block_N, in_dtype, out_dtype, accum_dtype)
kernel = TL.compile(gemm, out_idx=[]) kernel = TL.compile(gemm, out_idx=[])
src_code = kernel.get_kernel_source() src_code = kernel.get_kernel_source()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment