Skip to content

Commit

Permalink
indent & readme
Browse files Browse the repository at this point in the history
  • Loading branch information
wxj6000 committed Jan 29, 2024
1 parent 7c71037 commit 3ca0a23
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions gpu4pyscf/lib/cutensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,14 +55,16 @@ def _create_tensor_descriptor(a):
return _tensor_descriptors[key]

def contraction(
pattern, a, b, alpha, beta, out=None,
pattern, a, b, alpha, beta,
out=None,
op_a=cutensor_backend.OP_IDENTITY,
op_b=cutensor_backend.OP_IDENTITY,
op_c=cutensor_backend.OP_IDENTITY,
algo=cutensor_backend.ALGO_DEFAULT,
jit_mode=cutensor_backend.JIT_MODE_NONE,
compute_desc=0,
ws_pref=cutensor_backend.WORKSPACE_RECOMMENDED):
ws_pref=cutensor_backend.WORKSPACE_RECOMMENDED
):

pattern = pattern.replace(" ", "")
str_a, rest = pattern.split(',')
Expand Down

0 comments on commit 3ca0a23

Please sign in to comment.