intel / torch-xpu-ops

Apache License 2.0
22 stars 15 forks source link

CPU/CUDA bias code in aten::mode_out #327

Open fengyuan14 opened 3 months ago

fengyuan14 commented 3 months ago

🚀 The feature, motivation and pitch

Hard-coded CPU/CUDA bias in aten::mode_out. To upstream to make the operator device compatible.

Alternatives

No response

Additional context

No response

chuanqi129 commented 1 month ago

475 # CPU/CUDA bias code in aten::mode_out 1476 # https://github.com/intel/torch-xpu-ops/issues/327 1477 # RuntimeError: mode only supports CPU AND CUDA device type, got: xpu 1478 "test_dim_reduction", 1479 "test_mode", 1480 "test_dim_reduction_fns_fn_name_mode",