Update ck_gemm_template.h

ROCm · Oct 9, 2024 · 3033c62 · 3033c62
1 parent c1180e3
commit 3033c62
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/aten/src/ATen/native/hip/ck_gemm_template.h b/aten/src/ATen/native/hip/ck_gemm_template.h
@@ -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>