dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

wip

This commit is contained in:
louisfd 2024-07-13 14:28:56 -04:00
parent 32f05968b3
commit d0abd05395
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
crates/burn-cuda/src/compiler/shader.rs
View File

@ -87,9 +87,9 @@ impl CompilerRepresentation for ComputeShader {
impl Display for ComputeShader {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
if self.wmma_activated {
// if self.wmma_activated {
f.write_str("#include <mma.h>\nusing namespace nvcuda;\n")?;
}
//}
f.write_fmt(format_args!(
"