Skip to content

Commit 8a7aa34

Browse files
committed
replacing gpu_cpu with just gpu string for implementation
1 parent a5ee145 commit 8a7aa34

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

httomo_backends/methods_database/packages/backends/httomolibgpu/httomolibgpu.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ recon:
148148
FBP2d_astra:
149149
pattern: sinogram
150150
output_dims_change: True
151-
implementation: gpu_cpu
151+
implementation: gpu
152152
save_result_default: True
153153
padding: False
154154
memory_gpu:

httomo_backends/methods_database/query.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,13 +101,12 @@ def get_output_dims_change(self) -> bool:
101101
p = self._get_method_info("output_dims_change")
102102
return bool(p)
103103

104-
def get_implementation(self) -> Literal["cpu", "gpu", "gpu_cupy", "gpu_cpu"]:
104+
def get_implementation(self) -> Literal["cpu", "gpu", "gpu_cupy"]:
105105
p = self._get_method_info("implementation")
106106
assert p in [
107107
"gpu",
108108
"gpu_cupy",
109109
"cpu",
110-
"gpu_cpu",
111110
], f"The implementation arch {p} listed for method {self.module_path}.{self.method_name} is invalid"
112111
return p
113112

tests/test_method_query.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def test_httomolibgpu_implementation():
4343

4444
def test_httomolibgpu_implementation2():
4545
query = MethodsDatabaseQuery("httomolibgpu.recon.algorithm", "FBP2d_astra")
46-
assert query.get_implementation() == "gpu_cpu"
46+
assert query.get_implementation() == "gpu"
4747

4848

4949
def test_httomolibgpu_output_dims_change():

0 commit comments

Comments
 (0)