Comments (2)
I just ran the following code based on your example and it worked properly for me. Maybe you could provide more details on how to reproduce the issue?
fn main() -> Result<()> {
let dev = Device::new_cuda(0)?;
let conv_config = Conv2dConfig {
stride: 14,
..Default::default()
};
let vb = VarBuilder::zeros(DType::BF16, &dev);
let patch_embedding = conv2d(
/* in= */ 3,
/* out= */ 1152,
/* kernel_size = */ 14,
conv_config,
vb.pp("patch_embedding"),
)?;
let t = Tensor::zeros((5, 3, 663, 980), DType::BF16, &dev)?;
let res = patch_embedding.forward(&t)?;
println!("{res}");
Ok(())
}
from candle.
I think I figured it out. If we set
candle_core::cuda::set_gemm_reduced_precision_f16(true);
Then it crashes. I think this is not a bug though, so I'll close the issue.
from candle.
Related Issues (20)
- Status of Apple silicon M3 GPU support?
- Quantized-t5 models on Cuda HOT 2
- WASM library examples require HTTPS
- How to select which GPU to use HOT 3
- Metal memory leak multiplying matrices HOT 3
- Is it advisable to avoid variable shadowing when using Candle?
- candle-flash-attn infinite compile time HOT 1
- Model Context Not Resetting Between Requests Leading to Invalid Responses
- Tensor not dropped after another tensor was assigned to variable in cuda HOT 3
- Mamba model is broken with `f16` precision HOT 1
- Slow YOLOv8 using MKL
- candle examples fail on cuda with CUDA_DEVICE_NO_ERROR. Device caps of 7.5 HOT 2
- Parler TTS support?
- Positional encoding in DINOv2: Case without interpolation
- new model : DINOv2 reg4 with weights from PlantCLEF 2024 HOT 1
- Retrieving softmax_lse in candle-flash-attn HOT 1
- load_image and load_image_and_resize return different permutations
- How to get raw tensor data? HOT 2
- Qwen/Qwen2-7B doesn't work properly in the example qwen
- Gemma 2 support HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from candle.