diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-22 16:06:35 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-22 15:06:35 +0100 |
| commit | c8459d199ddcea909f6ccd18ae4945cb19d3eb9e (patch) | |
| tree | d7ce54723e63703a10a2b1efb6c794b8b4922895 /candle-kernels/build.rs | |
| parent | 1f2604269350cbba2409e54d413d68718ef36858 (diff) | |
| download | candle-c8459d199ddcea909f6ccd18ae4945cb19d3eb9e.tar.gz candle-c8459d199ddcea909f6ccd18ae4945cb19d3eb9e.tar.bz2 candle-c8459d199ddcea909f6ccd18ae4945cb19d3eb9e.zip | |
Add the layer norm files. (#222)
Diffstat (limited to 'candle-kernels/build.rs')
| -rw-r--r-- | candle-kernels/build.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-kernels/build.rs b/candle-kernels/build.rs index 3c8e96a9..58518412 100644 --- a/candle-kernels/build.rs +++ b/candle-kernels/build.rs @@ -184,6 +184,7 @@ mod cuda { let mut command = std::process::Command::new("nvcc"); command.arg(format!("--gpu-architecture=sm_{compute_cap}")) .arg("--ptx") + .arg("--expt-relaxed-constexpr") .args(["--default-stream", "per-thread"]) .args(["--output-directory", &out_dir]) // Flash attention only |