diff --git a/crates/ratchet-core/src/ops/select.rs b/crates/ratchet-core/src/ops/select.rs index e11f37a6..623e2817 100644 --- a/crates/ratchet-core/src/ops/select.rs +++ b/crates/ratchet-core/src/ops/select.rs @@ -356,8 +356,9 @@ def index_select(input, indices): let device = Device::request_device(DeviceRequest::GPU).unwrap(); run_index_select_trial(prob.clone(), device, true); - let device = Device::request_device(DeviceRequest::CPU).unwrap(); - run_index_select_trial(prob, device, true); + // TODO: impl CPU index_select for quantized tensors + // let device = Device::request_device(DeviceRequest::CPU).unwrap(); + // run_index_select_trial(prob, device, true); } #[derive(Debug, Clone)]