/src/test/codegen/prefetch.rs

https://gitlab.com/jianglu/rust · Rust · 73 lines · 40 code · 7 blank · 26 comment · 0 complexity · 7b72df8ed908de67e21cf436b9e5e788 MD5 · raw file 

    

  1. // Copyright 2017 The Rust Project Developers. See the COPYRIGHT
    2. 

  2. // file at the top-level directory of this distribution and at
    3. 

  3. // http://rust-lang.org/COPYRIGHT.
    4. 

  4. //
    5. 

  5. // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
    6. 

  6. // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
    7. 

  7. // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
    8. 

  8. // option. This file may not be copied, modified, or distributed
    9. 

  9. // except according to those terms.
    10. 

  10. 

  11. // compile-flags: -C no-prepopulate-passes
    12. 

  12. 

  13. #![crate_type = "lib"]
    14. 

  14. #![feature(core_intrinsics)]
    15. 

  15. 

  16. use std::intrinsics::{prefetch_read_data, prefetch_write_data,
    17. 

  17.                       prefetch_read_instruction, prefetch_write_instruction};
    18. 

  18. 

  19. #[no_mangle]
    20. 

  20. pub fn check_prefetch_read_data(data: &[i8]) {
    21. 

  21.     unsafe {
    22. 

  22.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 0, i32 0, i32 1)
    23. 

  23.         prefetch_read_data(data.as_ptr(), 0);
    24. 

  24.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 0, i32 1, i32 1)
    25. 

  25.         prefetch_read_data(data.as_ptr(), 1);
    26. 

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

  27.         prefetch_read_data(data.as_ptr(), 2);
    28. 

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

  29.         prefetch_read_data(data.as_ptr(), 3);
    30. 

  30.     }
    31. 

  31. }
    32. 

  32. 

  33. #[no_mangle]
    34. 

  34. pub fn check_prefetch_write_data(data: &[i8]) {
    35. 

  35.     unsafe {
    36. 

  36.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 1, i32 0, i32 1)
    37. 

  37.         prefetch_write_data(data.as_ptr(), 0);
    38. 

  38.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 1, i32 1, i32 1)
    39. 

  39.         prefetch_write_data(data.as_ptr(), 1);
    40. 

  40.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 1, i32 2, i32 1)
    41. 

  41.         prefetch_write_data(data.as_ptr(), 2);
    42. 

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

  43.         prefetch_write_data(data.as_ptr(), 3);
    44. 

  44.     }
    45. 

  45. }
    46. 

  46. 

  47. #[no_mangle]
    48. 

  48. pub fn check_prefetch_read_instruction(data: &[i8]) {
    49. 

  49.     unsafe {
    50. 

  50.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 0, i32 0, i32 0)
    51. 

  51.         prefetch_read_instruction(data.as_ptr(), 0);
    52. 

  52.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 0, i32 1, i32 0)
    53. 

  53.         prefetch_read_instruction(data.as_ptr(), 1);
    54. 

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

  55.         prefetch_read_instruction(data.as_ptr(), 2);
    56. 

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

  57.         prefetch_read_instruction(data.as_ptr(), 3);
    58. 

  58.     }
    59. 

  59. }
    60. 

  60. 

  61. #[no_mangle]
    62. 

  62. pub fn check_prefetch_write_instruction(data: &[i8]) {
    63. 

  63.     unsafe {
    64. 

  64.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 1, i32 0, i32 0)
    65. 

  65.         prefetch_write_instruction(data.as_ptr(), 0);
    66. 

  66.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 1, i32 1, i32 0)
    67. 

  67.         prefetch_write_instruction(data.as_ptr(), 1);
    68. 

  68.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 1, i32 2, i32 0)
    69. 

  69.         prefetch_write_instruction(data.as_ptr(), 2);
    70. 

  70.         // CHECK: call void @llvm.prefetch(i8* %{{.*}}, i32 1, i32 3, i32 0)
    71. 

  71.         prefetch_write_instruction(data.as_ptr(), 3);
    72. 

  72.     }
    73. 

  73. }
    74.