Search Options

Display Count
Sort
Preferred Language
Advanced Search

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

  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')):
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Fri Dec 19 16:32:04 GMT 2025
    - 48.3K bytes
    - Click Count (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"
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Mar 28 04:33:01 GMT 2026
    - 58.9K bytes
    - Click Count (0)
Back to Top