cuquantum.cutensornet.workspace_compute_gate_split_sizes

cuquantum.cutensornet.workspace_compute_gate_split_sizes(intptr_t handle, intptr_t tensor_a, intptr_t tensor_b, intptr_t tensor_g, intptr_t tensor_u, intptr_t tensor_v, int algo, intptr_t svd_config, int compute_type, intptr_t workspace)[source]

Compute the required workspace sizes for gate_split().

Parameters
  • handle (intptr_t) – The library handle.

  • tensor_a (intptr_t) – The tensor descriptor for the input A.

  • tensor_b (intptr_t) – The tensor descriptor for the input B.

  • tensor_g (intptr_t) – The tensor descriptor for the input G (the gate).

  • tensor_u (intptr_t) – The tensor descriptor for the output U.

  • tensor_v (intptr_t) – The tensor descriptor for the output V.

  • algo (cuquantum.cutensornet.GateSplitAlgo) – The gate splitting algorithm.

  • svd_config (intptr_t) – The tensor SVD config handle.

  • compute_type (cuquantum.ComputeType) – The compute type of the computation.

  • workspace (intptr_t) – The workspace descriptor.