File tree 2 files changed +2
-1
lines changed
2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -170,6 +170,7 @@ ifdef LLAMA_CUDA
170
170
OBJS += ggml-cuda.o
171
171
NVCC = nvcc
172
172
NVCCFLAGS = --forward-unknown-to-host-compiler
173
+ NVCCV := $(shell $(NVCC) --version | tail -n 1)
173
174
ifdef LLAMA_DEBUG
174
175
NVCCFLAGS += -lineinfo
175
176
endif # LLAMA_DEBUG
Original file line number Diff line number Diff line change @@ -14,7 +14,7 @@ template<> struct vec2_t_impl<float> { typedef float2 type; };
14
14
template <typename T> using vec2_t = typename vec2_t_impl<T>::type;
15
15
16
16
template <typename T> inline __host__ __device__ vec2_t <T> make_vec2_t (const T & x, const T & y);
17
- template <> inline __host__ __device__ vec2_t <half> make_vec2_t (const half & x, const half & y) { return __halves2half2 (x, y); }
17
+ template <> inline __host__ __device__ vec2_t <half> make_vec2_t (const half & x, const half & y) { return make_half2 (x, y); }
18
18
template <> inline __host__ __device__ vec2_t <float > make_vec2_t (const float & x, const float & y) { return make_float2 (x, y); }
19
19
20
20
// the cuda headers define operators for half2, but not for float2
You can’t perform that action at this time.
0 commit comments