forked from ccf-ai-infra/GPUCodeForces
This commit is contained in:
parent
7363ea3663
commit
bed466e72a
|
|
@ -110,7 +110,7 @@ canberra_focal = load_inline(
|
|||
cpp_sources=canberra_focal_cpp_source,
|
||||
cuda_sources=canberra_focal_source,
|
||||
functions=["canberra_focal_loss_cuda"],
|
||||
extra_cuda_cflags=["-O3", "--use_fast_math"],
|
||||
extra_cuda_cflags=["-O3"],
|
||||
verbose=True
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue