summaryrefslogtreecommitdiff
path: root/candle-core/src/tensor.rs
diff options
context:
space:
mode:
authoryinqiwen <yinqiwen@gmail.com>2024-03-23 20:44:10 +0800
committerGitHub <noreply@github.com>2024-03-23 13:44:10 +0100
commit790037390ca81a9cf32f35c03f514452d1366e4f (patch)
tree899801ed338767776303e8581ebedfe204522453 /candle-core/src/tensor.rs
parent6f877592a7d5b5023462e0b8d241a2ba5ad83648 (diff)
downloadcandle-790037390ca81a9cf32f35c03f514452d1366e4f.tar.gz
candle-790037390ca81a9cf32f35c03f514452d1366e4f.tar.bz2
candle-790037390ca81a9cf32f35c03f514452d1366e4f.zip
Add cast_bf16_x/cast_x_bf16 when CUDA_ARCH<800 but CUDA_VERSION >= 11000 (#1919)
- it make possible to load bf16 models on T4(sm75)
Diffstat (limited to 'candle-core/src/tensor.rs')
0 files changed, 0 insertions, 0 deletions