Skip to content

Commit 455294d

Browse files
authored
bugfix: Add guard for fp4/fp8 related include headers (#1376)
<!-- .github/pull_request_template.md --> ## ๐Ÿ“Œ Description Fix compilation issue of `cutlass_utils.cuh` ## ๐Ÿ” Related Issues <!-- Link any related issues here --> ## ๐Ÿš€ Pull Request Checklist Thank you for contributing to FlashInfer! Before we review your pull request, please make sure the following items are complete. ### โœ… Pre-commit Checks - [x] I have installed `pre-commit` by running `pip install pre-commit` (or used your preferred method). - [x] I have installed the hooks with `pre-commit install`. - [x] I have run the hooks manually with `pre-commit run --all-files` and fixed any reported issues. > If you are unsure about how to set up `pre-commit`, see [the pre-commit documentation](https://pre-commit.com/). ## ๐Ÿงช Tests - [x] Tests have been added or updated as needed. - [x] All tests are passing (`unittest`, etc.). ## Reviewer Notes <!-- Optional: anything you'd like reviewers to focus on, concerns, etc. -->
1 parent b79b78c commit 455294d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

โ€Žinclude/flashinfer/cutlass_utils.cuh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
#ifndef FLASHINFER_CUTLASS_UTILS_CUH_
1717
#define FLASHINFER_CUTLASS_UTILS_CUH_
1818

19+
#include <cuda_fp8.h>
20+
1921
#include "cute/tensor.hpp"
2022
#include "cutlass/cutlass.h"
2123
#include "cutlass/epilogue/collective/collective_builder.hpp"
@@ -39,7 +41,6 @@
3941
#include "cutlass/util/reference/device/tensor_compare.h"
4042
#include "cutlass/util/reference/device/tensor_fill.h"
4143
#include "cutlass/util/tensor_view_io.h"
42-
4344
#if defined(FLASHINFER_ENABLE_FP4_E2M1)
4445
#include <cuda_fp4.h>
4546
#endif

0 commit comments

Comments
ย (0)