Cargo.toml

[package]
name = "candle-cublaslt"
version = "0.2.2"
edition = "2021"

description = "CUBLASLt gemm for the candle ML framework."

[dependencies]
#candle = { version = "*", package = "candle-core", features = ["cuda"]}
candle = { git = "ssh://git@developer.sourcefind.cn:yongshk/candle.git", branch = "update_flash_nodebug_cutlass",  package = "candle-core", features = ["cuda"]}
cudarc = { git = "ssh://git@developer.sourcefind.cn:yongshk/cudarc.git", branch = "specific-main-nodebug", default-features = false, features = [ "cublaslt", "f16" ]}
#cudarc = { git = "https://github.com/coreylowman/cudarc", rev = "c388e724af93a3e8fbe484f5ded2d8b3c1badd8e", default-features = false, features = [ "cublaslt", "f16" ]}
half = { version = "2.3.1", features = ["num-traits"] }

[patch.crates-io]
#cudarc = { git = "ssh://git@developer.sourcefind.cn:yongshk/cudarc.git", branch = "specific-main-nodebug", default-features = false, features = [ "cublaslt", "f16" ]}
cudarc = { git = "https://github.com/coreylowman/cudarc", rev = "c388e724af93a3e8fbe484f5ded2d8b3c1badd8e", default-features = false, features = [ "cublaslt", "f16" ]}

[features]
default = []
static-linking = ["cudarc/static-linking"]