Skip to content

Commit 47a61d5

Browse files
committed
Make lowlatencygemm specifically point to the required include call
Signed-off-by: Phillip Kuznetsov <[email protected]>
1 parent c1fd461 commit 47a61d5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

cpp/tensorrt_llm/plugins/lowLatencyGemmPlugin/lowLatencyGemmPlugin.h

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,9 @@
1717

1818
#pragma once
1919

20-
#include "low_latency_gemm.h"
20+
// Specified this path instead of just the file name to disambiguate "opened cutlass" from "internal cutlass"
21+
// And this file specifically uses "internal cutlass"
22+
#include "tensorrt_llm/kernels/internal_cutlass_kernels/include/low_latency_gemm.h"
2123

2224
#include "tensorrt_llm/plugins/common/gemmPluginProfiler.h"
2325
#include "tensorrt_llm/plugins/common/plugin.h"

0 commit comments

Comments
 (0)