Skip to content

GitLab

Commit 485e530b authored by Jiming Ruan's avatar Jiming Ruan
Browse files

adjust workload

parent 15324083
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
example/ck_tile/10_rmsnorm2d/generate.py
View file @ 485e530b
...@@ -536,7 +536,9 @@ float rmsnorm2d_fwd(rmsnorm2d_fwd_traits t, ...@@ -536,7 +536,9 @@ float rmsnorm2d_fwd(rmsnorm2d_fwd_traits t,
h_traits('x', 'y', 'xs', 'ys', 1, 4, 1,1024, 2, True, False, False, 0, 0), h_traits('x', 'y', 'xs', 'ys', 1, 4, 1,1024, 2, True, False, False, 0, 0),
h_traits('x', 'y', 'xs', 'ys', 1, 8, 1,1024, 1, True, False, False, 0, 0)], h_traits('x', 'y', 'xs', 'ys', 1, 8, 1,1024, 1, True, False, False, 0, 0)],
'big' :[ h_traits('x', 'y', 'xs', 'ys', 1, 1, 1,1024, 8, True, False, True, 0, 0), 'big' :[ h_traits('x', 'y', 'xs', 'ys', 1, 1, 1,1024, 8, True, False, True, 0, 0),
h_traits('x', 'y', 'xs', 'ys', 1, 20, 1, 256, 1, True, False, True, 0, 0)]} h_traits('x', 'y', 'xs', 'ys', 1, 4, 1, 256, 4, True, False, True, 0, 0),
h_traits('x', 'y', 'xs', 'ys', 1, 12, 1, 256, 2, True, False, True, 0, 0),
h_traits('x', 'y', 'xs', 'ys', 1, 16, 1, 256, 1, True, False, True, 0, 0)]}
total_blob = list() total_blob = list()
for hs_key in h_trait_dict: for hs_key in h_trait_dict:
hs = h_trait_dict[hs_key] hs = h_trait_dict[hs_key]
......
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