kernels-community
/

quantization

Model card Files Files and versions Community

danieldk HF Staff commited on 8 days ago

Commit

85bad96

·

1 Parent(s): 9dfcd53

Temporarily disable ROCm

Files changed (1) hide show

build.toml +2 -3

build.toml CHANGED Viewed

@@ -46,8 +46,8 @@ include = [ "." ]
 depends = [ "cutlass_3_6", "torch" ]
 [kernel.fp8_common]
-language = "cuda-hipify"
-rocm-archs = [ "gfx906", "gfx908", "gfx90a", "gfx940", "gfx941", "gfx942", "gfx1030", "gfx1100", "gfx1101" ]
 src = [
   "fp8/amd/hip_float8.h",
   "fp8/amd/hip_float8_impl.h",
@@ -69,7 +69,6 @@ src = [
 depends = [ "torch" ]
 [kernel.int8_common]
-language = "cuda-hipify"
 cuda-capabilities = [ "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0" ]
 rocm-archs = [ "gfx906", "gfx908", "gfx90a", "gfx940", "gfx941", "gfx942", "gfx1030", "gfx1100", "gfx1101" ]
 src = [

 depends = [ "cutlass_3_6", "torch" ]
 [kernel.fp8_common]
+# language = "cuda-hipify"
+# rocm-archs = [ "gfx906", "gfx908", "gfx90a", "gfx940", "gfx941", "gfx942", "gfx1030", "gfx1100", "gfx1101" ]
 src = [
   "fp8/amd/hip_float8.h",
   "fp8/amd/hip_float8_impl.h",
 depends = [ "torch" ]
 [kernel.int8_common]
 cuda-capabilities = [ "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0" ]
 rocm-archs = [ "gfx906", "gfx908", "gfx90a", "gfx940", "gfx941", "gfx942", "gfx1030", "gfx1100", "gfx1101" ]
 src = [