rich7420 commented on code in PR #747:
URL: https://github.com/apache/mahout/pull/747#discussion_r2637511700
##########
qdp/qdp-core/tests/api_workflow.rs:
##########
@@ -107,3 +107,103 @@ fn test_amplitude_encoding_async_pipeline() {
println!("PASS: Memory freed successfully");
}
}
+
+#[test]
+#[cfg(target_os = "linux")]
+fn test_batch_dlpack_2d_shape() {
+ println!("Testing batch DLPack 2D shape...");
+
+ let engine = match QdpEngine::new(0) {
+ Ok(e) => e,
+ Err(_) => {
+ println!("SKIP: No GPU available");
+ return;
+ }
+ };
+
+ // Create batch data: 3 samples, each with 4 elements (2 qubits)
+ let num_samples = 3;
+ let num_qubits = 2;
+ let sample_size = 4;
+ let batch_data: Vec<f64> = (0..num_samples * sample_size)
+ .map(|i| (i as f64) / 10.0)
+ .collect();
+
+ let result = engine.encode_batch(&batch_data, num_samples, sample_size,
num_qubits, "amplitude");
Review Comment:
you're right! let me fix it
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
