@@ -249,20 +249,6 @@ mod tests {
249249 output. into_data ( ) . assert_eq ( & expected, false ) ;
250250 }
251251
252- #[ test]
253- fn should_select_assign_bool_empty_indices ( ) {
254- // Test empty indices array
255- let device = Default :: default ( ) ;
256- let tensor = TestTensorBool :: < 1 > :: from_data ( [ true , false , true ] , & device) ;
257- let indices = TestTensorInt :: < 1 > :: from_data ( [ ] as [ i32 ; 0 ] , & device) ;
258- let values = TestTensorBool :: < 1 > :: from_data ( [ ] as [ bool ; 0 ] , & device) ;
259-
260- let output = tensor. select_assign ( 0 , indices, values) ;
261- let expected = TensorData :: from ( [ true , false , true ] ) ;
262-
263- output. into_data ( ) . assert_eq ( & expected, false ) ;
264- }
265-
266252 #[ test]
267253 fn should_select_assign_bool_true_or_true_accumulation ( ) {
268254 // Test multiple true accumulations
@@ -280,8 +266,6 @@ mod tests {
280266 #[ test]
281267 fn should_match_default_implementation_behavior ( ) {
282268 // Verify optimized implementation matches original default logic
283- use burn_tensor:: backend:: Backend ;
284-
285269 let device = Default :: default ( ) ;
286270 let tensor = TestTensorBool :: < 1 > :: from_data ( [ true , false , true ] , & device) ;
287271 let indices = TestTensorInt :: from_data ( [ 0 , 1 , 0 ] , & device) ;
@@ -305,8 +289,6 @@ mod tests {
305289 #[ test]
306290 fn should_select_assign_bool_overlapping_indices_vs_default ( ) {
307291 // Test overlapping indices against default implementation
308- use burn_tensor:: backend:: Backend ;
309-
310292 let device = Default :: default ( ) ;
311293 let tensor = TestTensorBool :: < 1 > :: from_data ( [ false , true ] , & device) ;
312294 let indices = TestTensorInt :: from_data ( [ 0 , 0 ] , & device) ;
@@ -329,8 +311,6 @@ mod tests {
329311 #[ test]
330312 fn should_select_assign_bool_true_or_true_accumulation_vs_default ( ) {
331313 // Test multiple true accumulations against default implementation
332- use burn_tensor:: backend:: Backend ;
333-
334314 let device = Default :: default ( ) ;
335315 let tensor = TestTensorBool :: < 1 > :: from_data ( [ true , false ] , & device) ;
336316 let indices = TestTensorInt :: from_data ( [ 0 , 0 , 0 ] , & device) ;
@@ -374,30 +354,6 @@ mod tests {
374354 . assert_eq ( & default_result. into_data ( ) , false ) ;
375355 }
376356
377- #[ test]
378- fn should_select_assign_bool_empty_indices_vs_default ( ) {
379- // Test empty indices against default implementation
380- use burn_tensor:: backend:: Backend ;
381-
382- let device = Default :: default ( ) ;
383- let tensor = TestTensorBool :: < 1 > :: from_data ( [ true , false , true ] , & device) ;
384- let indices = TestTensorInt :: < 1 > :: from_data ( [ ] as [ i32 ; 0 ] , & device) ;
385- let values = TestTensorBool :: < 1 > :: from_data ( [ ] as [ bool ; 0 ] , & device) ;
386-
387- let optimized_result = tensor
388- . clone ( )
389- . select_assign ( 0 , indices. clone ( ) , values. clone ( ) ) ;
390-
391- let int_tensor = tensor. int ( ) ;
392- let int_values = values. int ( ) ;
393- let assigned = int_tensor. select_assign ( 0 , indices, int_values) ;
394- let default_result = assigned. greater_elem ( 0 ) ;
395-
396- optimized_result
397- . into_data ( )
398- . assert_eq ( & default_result. into_data ( ) , false ) ;
399- }
400-
401357 #[ test]
402358 fn should_select_assign_bool_tensor_vs_default ( ) {
403359 // Test existing basic case against default implementation
0 commit comments