Skip to content

Commit

Permalink
Fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
EricLBuehler committed Aug 18, 2024
1 parent cf2494b commit ca4b0d6
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion candle-core/src/safetensors.rs
Original file line number Diff line number Diff line change
Expand Up @@ -206,7 +206,7 @@ fn convert(view: &st::TensorView<'_>, device: &Device) -> Result<Tensor> {
convert_with_cast_::<u16, u32, _>(view, device, conv)
}
st::Dtype::U32 => convert_::<u32>(view, device),
st::Dtype::I32 => onvert_::<i32>(view, device),
st::Dtype::I32 => convert_::<i32>(view, device),
st::Dtype::I64 => convert_::<i64>(view, device),
st::Dtype::BF16 => convert_::<half::bf16>(view, device),
st::Dtype::F16 => convert_::<half::f16>(view, device),
Expand Down
2 changes: 1 addition & 1 deletion candle-kernels/src/indexing.cu
Original file line number Diff line number Diff line change
Expand Up @@ -199,7 +199,7 @@ IS_OP(int64_t, int64_t, is_i64_i64)
IS_OP(int32_t, int64_t, is_i64_i32)

IS_OP(float, uint32_t, is_u32_f32)
IS_OP(double, uint32_t, is_u32_f64)w
IS_OP(double, uint32_t, is_u32_f64)
IS_OP(uint8_t, uint32_t, is_u32_u8)
IS_OP(int32_t, uint32_t, is_u32_i32)
IS_OP(int64_t, uint32_t, is_u32_i64)
Expand Down

0 comments on commit ca4b0d6

Please sign in to comment.