Skip to content

GitLab

Commit ef9eae46 authored by yongshk's avatar yongshk
Browse files

Initial commit

parent 443bbca9
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
Cargo.toml
View file @ ef9eae46
......@@ -7,13 +7,13 @@ 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" ]}
candle = { git = "http://developer.sourcefind.cn/codes/yongshk/candle", branch = "update_flash_nodebug_cutlass", package = "candle-core", features = ["cuda"]}
cudarc = { git = "http://developer.sourcefind.cn/codes/yongshk/cudarc", 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 = "http://developer.sourcefind.cn/codes/yongshk/cudarc", 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]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment