Skip to content

GitLab

Commit 3dcd7dc0 authored by Vijay Korthikanti's avatar Vijay Korthikanti
Browse files

build directory

parent a7f4e460
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
megatron/fused_kernels/__init__.py
View file @ 3dcd7dc0
......@@ -51,7 +51,7 @@ def load_scaled_upper_triang_masked_softmax_fusion_kernel():
name='scaled_upper_triang_masked_softmax_cuda',
sources=[srcpath / 'scaled_upper_triang_masked_softmax.cpp',
srcpath / 'scaled_upper_triang_masked_softmax_cuda.cu'],
build_directory=srcpath,
build_directory=srcpath / 'build',
extra_cflags=['-O3',],
extra_cuda_cflags=['-O3',
'-gencode', 'arch=compute_70,code=sm_70',
......@@ -76,7 +76,7 @@ def load_scaled_masked_softmax_fusion_kernel():
name='scaled_masked_softmax_cuda',
sources=[srcpath / 'scaled_masked_softmax.cpp',
srcpath / 'scaled_masked_softmax_cuda.cu'],
build_directory=srcpath,
build_directory=srcpath / 'build',
extra_cflags=['-O3',],
extra_cuda_cflags=['-O3',
'-gencode', 'arch=compute_70,code=sm_70',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment