Skip to content

GitLab

"tests/pipelines/vscode:/vscode.git/clone" did not exist on "0cbefefac3363666ea2f1b1f730a019214a8b3d4"
Unverified Commit c6a19fb2 authored by Leon Lu's avatar Leon Lu Committed by GitHub
Browse files

[Bugfix] Fix the jit_kernel issue (#1357) 



* [Bugfix] Fix the jit_kernel issue

* Update README.md

---------
Co-authored-by: default avatarLei Wang <34334180+LeiWang1999@users.noreply.github.com>
parent a4ea7da9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
README.md
View file @ c6a19fb2
...@@ -209,7 +209,7 @@ torch.testing.assert_close(c, ref_c, rtol=1e-2, atol=1e-2) ...@@ -209,7 +209,7 @@ torch.testing.assert_close(c, ref_c, rtol=1e-2, atol=1e-2)
print("Kernel output matches PyTorch reference.") print("Kernel output matches PyTorch reference.")
# 4. Retrieve and inspect the generated CUDA source (optional) # 4. Retrieve and inspect the generated CUDA source (optional)
# cuda_source = jit_kernel.get_kernel_source() # cuda_source = matmul_relu_kernel.get_kernel_source()
# print("Generated CUDA kernel:\n", cuda_source) # print("Generated CUDA kernel:\n", cuda_source)
# 5.Profile latency with kernel # 5.Profile latency with kernel
......
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