/src/test/codegen/intrinsics/prefetch.rs

https://gitlab.com/rust-lang/rust · Rust · 63 lines · 40 code · 6 blank · 17 comment · 0 complexity · 5b321c74dd1693b57ca091c12b813036 MD5 · raw file 

    

  1. // compile-flags: -C no-prepopulate-passes
    2. 

  2. 

  3. #![crate_type = "lib"]
    4. 

  4. #![feature(core_intrinsics)]
    5. 

  5. 

  6. use std::intrinsics::{prefetch_read_data, prefetch_write_data,
    7. 

  7.                       prefetch_read_instruction, prefetch_write_instruction};
    8. 

  8. 

  9. #[no_mangle]
    10. 

  10. pub fn check_prefetch_read_data(data: &[i8]) {
    11. 

  11.     unsafe {
    12. 

  12.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 0, i32 1)
    13. 

  13.         prefetch_read_data(data.as_ptr(), 0);
    14. 

  14.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 1, i32 1)
    15. 

  15.         prefetch_read_data(data.as_ptr(), 1);
    16. 

  16.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 2, i32 1)
    17. 

  17.         prefetch_read_data(data.as_ptr(), 2);
    18. 

  18.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 3, i32 1)
    19. 

  19.         prefetch_read_data(data.as_ptr(), 3);
    20. 

  20.     }
    21. 

  21. }
    22. 

  22. 

  23. #[no_mangle]
    24. 

  24. pub fn check_prefetch_write_data(data: &[i8]) {
    25. 

  25.     unsafe {
    26. 

  26.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 0, i32 1)
    27. 

  27.         prefetch_write_data(data.as_ptr(), 0);
    28. 

  28.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 1, i32 1)
    29. 

  29.         prefetch_write_data(data.as_ptr(), 1);
    30. 

  30.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 2, i32 1)
    31. 

  31.         prefetch_write_data(data.as_ptr(), 2);
    32. 

  32.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 3, i32 1)
    33. 

  33.         prefetch_write_data(data.as_ptr(), 3);
    34. 

  34.     }
    35. 

  35. }
    36. 

  36. 

  37. #[no_mangle]
    38. 

  38. pub fn check_prefetch_read_instruction(data: &[i8]) {
    39. 

  39.     unsafe {
    40. 

  40.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 0, i32 0)
    41. 

  41.         prefetch_read_instruction(data.as_ptr(), 0);
    42. 

  42.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 1, i32 0)
    43. 

  43.         prefetch_read_instruction(data.as_ptr(), 1);
    44. 

  44.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 2, i32 0)
    45. 

  45.         prefetch_read_instruction(data.as_ptr(), 2);
    46. 

  46.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 0, i32 3, i32 0)
    47. 

  47.         prefetch_read_instruction(data.as_ptr(), 3);
    48. 

  48.     }
    49. 

  49. }
    50. 

  50. 

  51. #[no_mangle]
    52. 

  52. pub fn check_prefetch_write_instruction(data: &[i8]) {
    53. 

  53.     unsafe {
    54. 

  54.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 0, i32 0)
    55. 

  55.         prefetch_write_instruction(data.as_ptr(), 0);
    56. 

  56.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 1, i32 0)
    57. 

  57.         prefetch_write_instruction(data.as_ptr(), 1);
    58. 

  58.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 2, i32 0)
    59. 

  59.         prefetch_write_instruction(data.as_ptr(), 2);
    60. 

  60.         // CHECK: call void @llvm.prefetch{{.*}}({{.*}}, i32 1, i32 3, i32 0)
    61. 

  61.         prefetch_write_instruction(data.as_ptr(), 3);
    62. 

  62.     }
    63. 

  63. }
    64.