Skip to content

Commit

Permalink
Update symforce_cuda_codegen_test
Browse files Browse the repository at this point in the history
  • Loading branch information
matte1 committed Nov 13, 2024
1 parent 63c5cfc commit a0d0792
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion test/symforce_cuda_codegen_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ def cuda_func(
# Generate the symbolic backend test function
for scalar in scalars:
Codegen.function(
functools.partial(backend_test_function, [sf.sympy.loggamma, sf.sympy.erfc, sf.sympy.erf, sf.sympy.gamma]),
functools.partial(backend_test_function, [sf.sympy.erfc, sf.sympy.loggamma, sf.sympy.erf, sf.sympy.gamma]),
config=CudaConfig(inline=False, scalar_type=scalar),
name=f"backend_test_function_{scalar.value}",
).generate_function(output_dir, skip_directory_nesting=True)
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -226,11 +226,11 @@ __host__ __device__ void BackendTestFunctionFloat32(
}

if (res45 != nullptr) {
*res45 = lgammaf(x);
*res45 = erfcf(x);
}

if (res46 != nullptr) {
*res46 = erfcf(x);
*res46 = lgammaf(x);
}

if (res47 != nullptr) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -228,11 +228,11 @@ __host__ __device__ void BackendTestFunctionFloat64(
}

if (res45 != nullptr) {
*res45 = lgamma(x);
*res45 = erfc(x);
}

if (res46 != nullptr) {
*res46 = erfc(x);
*res46 = lgamma(x);
}

if (res47 != nullptr) {
Expand Down

0 comments on commit a0d0792

Please sign in to comment.