Skip to content

GitLab

Commit 349853d0 authored by fengzch's avatar fengzch
Browse files

fix: hip/hip_bf16.h not import

parent c78a5692
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/kernels/dispatch_utils.h
View file @ 349853d0
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
#include "common.h" #include "common.h"
#include "Tensor.h" #include "Tensor.h"
#include <cuda_fp16.h> #include <cuda_fp16.h>
#include <hip/hip_bf16.h>
template<typename F> template<typename F>
inline auto dispatchFloat(Tensor::ScalarType scalarType, F &&func) { inline auto dispatchFloat(Tensor::ScalarType scalarType, F &&func) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment