Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_NEED_ROCM (0.04 sec)

  1. configure.py

      set_action_env_var(
          environ_cp, 'TF_NEED_ROCM', 'ROCm', False, bazel_config_name='rocm')
      if (environ_cp.get('TF_NEED_ROCM') == '1' and
          'LD_LIBRARY_PATH' in environ_cp and
          environ_cp.get('LD_LIBRARY_PATH') != '1'):
        write_action_env_to_bazelrc('LD_LIBRARY_PATH',
                                    environ_cp.get('LD_LIBRARY_PATH'))
    
      if (environ_cp.get('TF_NEED_ROCM') == '1' and environ_cp.get('ROCM_PATH')):
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Wed Apr 30 15:18:54 UTC 2025
    - 48.3K bytes
    - Viewed (0)
  2. .bazelrc

    common:rocm --define=using_rocm_hipcc=true
    common:rocm --define=tensorflow_mkldnn_contraction_kernel=0
    common:rocm --define=xnn_enable_avxvnniint8=false
    common:rocm --define=xnn_enable_avx512fp16=false
    common:rocm --repo_env TF_NEED_ROCM=1
    
    common:rocm_clang_official --config=rocm
    common:rocm_clang_official --action_env=CLANG_COMPILER_PATH="/usr/lib/llvm-18/bin/clang"
    common:rocm_clang_official --action_env=TF_ROCM_CLANG="1"
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Fri Dec 26 23:20:26 UTC 2025
    - 56.8K bytes
    - Viewed (0)
Back to top