Kernels
danieldk HF Staff commited on
Commit
90f5a60
·
1 Parent(s): b9947d6

Disable ROCm for now (don't think it was ever tested)

Browse files
Files changed (1) hide show
  1. build.toml +24 -24
build.toml CHANGED
@@ -23,27 +23,27 @@ src = [
23
  include = [ "causal-conv1d" ]
24
  depends = [ "torch" ]
25
 
26
- [kernel.causal_conv1d_rocm]
27
- backend = "rocm"
28
- rocm-archs = [
29
- "gfx906",
30
- "gfx908",
31
- "gfx90a",
32
- "gfx940",
33
- "gfx941",
34
- "gfx942",
35
- "gfx1030",
36
- "gfx1100",
37
- "gfx1101",
38
- ]
39
- src = [
40
- "causal-conv1d/causal_conv1d_bwd.cu",
41
- "causal-conv1d/causal_conv1d_common.h",
42
- "causal-conv1d/causal_conv1d.cpp",
43
- "causal-conv1d/causal_conv1d_fwd.cu",
44
- "causal-conv1d/causal_conv1d.h",
45
- "causal-conv1d/causal_conv1d_update.cu",
46
- "causal-conv1d/static_switch.h",
47
- ]
48
- include = [ "causal-conv1d" ]
49
- depends = [ "torch" ]
 
23
  include = [ "causal-conv1d" ]
24
  depends = [ "torch" ]
25
 
26
+ #[kernel.causal_conv1d_rocm]
27
+ #backend = "rocm"
28
+ #rocm-archs = [
29
+ # "gfx906",
30
+ # "gfx908",
31
+ # "gfx90a",
32
+ # "gfx940",
33
+ # "gfx941",
34
+ # "gfx942",
35
+ # "gfx1030",
36
+ # "gfx1100",
37
+ # "gfx1101",
38
+ #]
39
+ #src = [
40
+ # "causal-conv1d/causal_conv1d_bwd.cu",
41
+ # "causal-conv1d/causal_conv1d_common.h",
42
+ # "causal-conv1d/causal_conv1d.cpp",
43
+ # "causal-conv1d/causal_conv1d_fwd.cu",
44
+ # "causal-conv1d/causal_conv1d.h",
45
+ # "causal-conv1d/causal_conv1d_update.cu",
46
+ # "causal-conv1d/static_switch.h",
47
+ #]
48
+ #include = [ "causal-conv1d" ]
49
+ #depends = [ "torch" ]