Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update ck_gemm_template.h
Browse files Browse the repository at this point in the history
pruthvistony authored Oct 9, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent c1180e3 commit 3033c62
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions aten/src/ATen/native/hip/ck_gemm_template.h
Original file line number Diff line number Diff line change
@@ -12,10 +12,8 @@

#include <ATen/ATen.h>
#include <ATen/hip/impl/HIPStreamMasqueradingAsCUDA.h>
#include <torch/torch.h>
#include <ATen/native/hip/ck_gemm.h>


#include <ck/ck.hpp>
#include <ck/tensor_operation/gpu/device/gemm_specialization.hpp>
#include <ck/tensor_operation/gpu/device/tensor_layout.hpp>

0 comments on commit 3033c62

Please sign in to comment.