kernels-community
/

layer_norm

Model card Files Files and versions

medmekk commited on Sep 18, 2025

Commit

70ebabf

·

1 Parent(s): 7fedc2c

update build

Files changed (1) hide show

build.toml +1 -0

build.toml CHANGED Viewed

@@ -79,6 +79,7 @@ src = [
     "layer-norm/ln_utils.cuh",
     "layer-norm/static_switch.h"
 ]
 cuda-flags = [
     "-O3",
     "-U__CUDA_NO_HALF_OPERATORS__",

     "layer-norm/ln_utils.cuh",
     "layer-norm/static_switch.h"
 ]
+cxx-flags = ["-DFLASHATTENTION_DISABLE_PYBIND"]
 cuda-flags = [
     "-O3",
     "-U__CUDA_NO_HALF_OPERATORS__",