From 13fa2004a14295ec84740e2ca40dbd2b9c13d0e2 Mon Sep 17 00:00:00 2001 From: Eric Buehler Date: Wed, 31 Jul 2024 20:45:33 -0400 Subject: [PATCH] Add bitwise op --- candle-kernels/src/bitwise.cu | 23 ----------------------- 1 file changed, 23 deletions(-) delete mode 100644 candle-kernels/src/bitwise.cu diff --git a/candle-kernels/src/bitwise.cu b/candle-kernels/src/bitwise.cu deleted file mode 100644 index c93f0e7aa..000000000 --- a/candle-kernels/src/bitwise.cu +++ /dev/null @@ -1,23 +0,0 @@ -#include "cuda_utils.cuh" -#include - -#define BITWISE_OP(TYPENAME, FN_NAME, OP) \ -extern "C" __global__ void FN_NAME(const TYPENAME *d_in1, const TYPENAME *d_in2, TYPENAME *d_out, \ - const uint32_t N) { \ - const int idx = blockIdx.x * blockDim.x + threadIdx.x; \ - if (idx < N) { \ - d_out[idx] = d_in1[idx] OP d_in2[idx]; \ - } \ -} \ - -BITWISE_OP(uint32_t, bitwise_or_u32, |) -BITWISE_OP(uint32_t, bitwise_and_u32, &) -BITWISE_OP(uint32_t, bitwise_xor_u32, ^) - -BITWISE_OP(uint8_t, bitwise_or_u8, |) -BITWISE_OP(uint8_t, bitwise_and_u8, &) -BITWISE_OP(uint8_t, bitwise_xor_u8, ^) - -BITWISE_OP(int64_t, bitwise_or_i64, |) -BITWISE_OP(int64_t, bitwise_and_i64, &) -BITWISE_OP(int64_t, bitwise_xor_i64, ^)