feat: update workspace paths and enhance gitignore
- Updated stablediffusion crate path from "../stable-diffusion-burn" to "./crates/stable-diffusion-burn" for proper workspace resolution - Enhanced .gitignore to include generated model files (.mpk, .pt, .bin, .safetensors, .ckpt) and user_data directory - Added Cargo.lock to gitignore with appropriate comment - Reorganized IDE files section in gitignore for better clarity - Added newline at end of file for proper formatting
This commit is contained in:
@@ -0,0 +1,42 @@
|
||||
[package]
|
||||
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
|
||||
categories = ["science"]
|
||||
description = "ROCm HIP backend for the Burn framework"
|
||||
edition.workspace = true
|
||||
keywords = ["deep-learning", "machine-learning", "gpu", "rocm", "hip"]
|
||||
license.workspace = true
|
||||
name = "burn-rocm"
|
||||
readme.workspace = true
|
||||
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-rocm"
|
||||
documentation = "https://docs.rs/burn-rocm"
|
||||
version.workspace = true
|
||||
|
||||
[lints]
|
||||
workspace = true
|
||||
|
||||
[features]
|
||||
default = ["fusion", "burn-cubecl/default", "cubecl/default"]
|
||||
tracing = [
|
||||
"cubecl/tracing",
|
||||
"burn-cubecl/tracing",
|
||||
"burn-backend/tracing",
|
||||
"burn-fusion?/tracing",
|
||||
]
|
||||
|
||||
fusion = ["burn-fusion", "burn-cubecl/fusion"]
|
||||
autotune = ["burn-cubecl/autotune"]
|
||||
autotune-checks = ["burn-cubecl/autotune-checks"]
|
||||
doc = ["burn-cubecl/doc"]
|
||||
std = ["burn-cubecl/std", "cubecl/std"]
|
||||
|
||||
[dependencies]
|
||||
cubecl = { workspace = true, features = ["hip"] }
|
||||
burn-cubecl = { path = "../burn-cubecl", version = "=0.21.0-pre.2", default-features = true }
|
||||
burn-backend = { path = "../burn-backend", version = "=0.21.0-pre.2", features = [
|
||||
"cubecl-hip",
|
||||
] }
|
||||
burn-fusion = { path = "../burn-fusion", version = "=0.21.0-pre.2", optional = true }
|
||||
|
||||
[package.metadata.docs.rs]
|
||||
features = ["doc"]
|
||||
rustdoc-args = ["--cfg", "docsrs"]
|
||||
@@ -0,0 +1,7 @@
|
||||
# burn-rocm
|
||||
|
||||
Backend using ROCm HIP runtime.
|
||||
|
||||
To execute the tests for this backend set an environment variable called `ROCM_PATH` or `CUBECL_ROCM_PATH` to the installation path of ROCm. It is often `/opt/rocm`.
|
||||
|
||||
For now this backend requires the version `6.2.2` of ROCm or a compatible version.
|
||||
@@ -0,0 +1,14 @@
|
||||
#![cfg_attr(docsrs, feature(doc_cfg))]
|
||||
extern crate alloc;
|
||||
|
||||
use burn_cubecl::CubeBackend;
|
||||
|
||||
pub use cubecl::hip::AmdDevice as RocmDevice;
|
||||
|
||||
use cubecl::hip::HipRuntime;
|
||||
|
||||
#[cfg(not(feature = "fusion"))]
|
||||
pub type Rocm<F = f32, I = i32, B = u8> = CubeBackend<HipRuntime, F, I, B>;
|
||||
|
||||
#[cfg(feature = "fusion")]
|
||||
pub type Rocm<F = f32, I = i32, B = u8> = burn_fusion::Fusion<CubeBackend<HipRuntime, F, I, B>>;
|
||||
Reference in New Issue
Block a user