Skip to content

Commit 9dadc55

Browse files
praguptafacebook-github-bot
authored andcommitted
Add flags to compile ck gemms under pytorch (pytorch#4878)
Summary: X-link: facebookresearch/FBGEMM#1903 Pull Request resolved: pytorch#4878 Reviewed By: ionuthristodorescu Differential Revision: D82511112 Pulled By: q10 fbshipit-source-id: 13f81cc7d19ff7202937e7c57d8bd898bf4c959e
1 parent ba6d456 commit 9dadc55

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_rowwise_grouped/fp8_rowwise_grouped_gemm.hip

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
* This source code is licensed under the BSD-style license found in the
66
* LICENSE file in the root directory of this source tree.
77
*/
8-
8+
#undef __HIP_NO_HALF_CONVERSIONS__
99
#include <cmath>
1010
#include <cstdlib>
1111
#include <string>

fbgemm_gpu/experimental/gen_ai/src/quantize/ck_extensions/fp8_rowwise_grouped/kernels/fp8_rowwise_grouped_common.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
* This source code is licensed under the BSD-style license found in the
66
* LICENSE file in the root directory of this source tree.
77
*/
8-
8+
#undef __HIP_NO_HALF_CONVERSIONS__
99
#include <ATen/ATen.h>
1010
#ifdef USE_ROCM
1111
#include <c10/hip/HIPStream.h>

0 commit comments

Comments
 (0)