Skip to content

Commit

Permalink
Only cast for now
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Jan 8, 2025
1 parent 261217d commit bba2c9a
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions candle-kernels/src/fill.cu
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,4 @@ COPY2D_OP(__nv_bfloat16, copy2d_bf16)

extern "C" __global__ void fill_f8_e4m3(__nv_fp8_e4m3 *buf, __nv_fp8_e4m3 value, const size_t numel) { fill_with(buf, value, numel); }
COPY2D_OP(__nv_fp8_e4m3, copy2d_f8_e4m3)
#else
extern "C" __global__ void fill_bf16(uint16_t *buf, uint16_t value, const size_t numel) { fill_with(buf, value, numel); }
COPY2D_OP(__nv_bfloat16, copy2d_bf16)
#endif

0 comments on commit bba2c9a

Please sign in to comment.