From 47de7eea9f12da0991db9e86b0c64cdec4678590 Mon Sep 17 00:00:00 2001 From: Nicholas Dahm Date: Fri, 14 Nov 2025 15:16:52 -0800 Subject: [PATCH 1/6] Added asmjit recipe --- recipes/asmjit/build.sh | 23 ++++++++++++++++ recipes/asmjit/meta.yaml | 58 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 81 insertions(+) create mode 100644 recipes/asmjit/build.sh create mode 100644 recipes/asmjit/meta.yaml diff --git a/recipes/asmjit/build.sh b/recipes/asmjit/build.sh new file mode 100644 index 0000000000000..3abf059037c92 --- /dev/null +++ b/recipes/asmjit/build.sh @@ -0,0 +1,23 @@ +#!/bin/bash + +set -ex + +mkdir build +cd build + +cmake ${CMAKE_ARGS} \ + -DCMAKE_BUILD_TYPE=Release \ + -DCMAKE_INSTALL_PREFIX=$PREFIX \ + -DCMAKE_INSTALL_LIBDIR=lib \ + -DASMJIT_STATIC=OFF \ + -DASMJIT_TEST=ON \ + -DASMJIT_NO_CUSTOM_FLAGS=OFF \ + .. + +cmake --build . --config Release --parallel ${CPU_COUNT} + +# Run tests to verify the build +# This must be done here, because otherwise we need to re-build the lib in test phase, which +ctest --output-on-failure --verbose + +cmake --install . --config Release diff --git a/recipes/asmjit/meta.yaml b/recipes/asmjit/meta.yaml new file mode 100644 index 0000000000000..d0e0805db36e7 --- /dev/null +++ b/recipes/asmjit/meta.yaml @@ -0,0 +1,58 @@ +{% set name = "asmjit" %} +{% set commit_date = "20251012" %} +{% set commit_hash = "5134d396bd00c1b63259387acdbb12dfdf009f9b" %} +{% set version = "0.0.0.dev" + commit_date + "+" + commit_hash[:8] %} + + +package: + name: {{ name|lower }} + version: {{ version }} + +source: + url: https://github.com/asmjit/asmjit/archive/{{ commit_hash }}.tar.gz + sha256: 7a138a9715ac52ac9c259818c05fe42dbf19233dbf82188640aaf32259fa1d34 + +build: + number: 0 + # TODO attempt to enable aarch64 after feedstock creation + # This needs c_stdlib_version>-2.28 + # In staged-recipes .ci_support/linux_aarch64.yaml pins c_stdlib_version=2.17 which fails this build + skip: true # [aarch64] + skip: true # [osx] + skip: true # [win] + +requirements: + build: + - {{ compiler('cxx') }} + - {{ stdlib('c') }} + - cmake >=3.24 + - make + +test: + commands: + # Verify installed files (tests were run at build time) + - test -f ${PREFIX}/include/asmjit/asmjit.h + - test -f ${PREFIX}/lib/libasmjit${SHLIB_EXT} + +about: + home: https://asmjit.com + summary: Low-latency machine code generation library + description: | + AsmJit is a lightweight library for machine code generation written in C++ language. + It can generate machine code for X86, X64, and ARM architectures with a focus on + low-latency code generation and JIT compilation. AsmJit provides: + - Complete x86/x64 instruction set with AVX-512 and AMX support + - Complete AArch64 instruction set including SVE/SVE2 + - Built-in CPU detection + - Advanced instruction validation + - Logging and error handling + - Virtual memory management + - Code generation performance comparable to hand-written assembly + license: Zlib + license_file: LICENSE.md + doc_url: https://asmjit.com/doc/index.html + dev_url: https://github.com/asmjit/asmjit + +extra: + recipe-maintainers: + - das-intensity From e2ae6b89af9cbcc58f532cc02d451d6c090dca9f Mon Sep 17 00:00:00 2001 From: Nicholas Dahm Date: Mon, 20 Oct 2025 22:03:34 -0700 Subject: [PATCH 2/6] Added fbgemm recipe building fbgemm, fbgemm-gpu, and fbgemm-gpu-genai --- recipes/fbgemm/LICENSE-asmjit.txt | 17 ++ recipes/fbgemm/LICENSE-cutlass.txt | 31 +++ recipes/fbgemm/LICENSE-fbgemm.txt | 30 +++ recipes/fbgemm/LICENSE-json.txt | 21 +++ recipes/fbgemm/LICENSE-quantize_ops_mx.txt | 21 +++ recipes/fbgemm/meta.yaml | 208 +++++++++++++++++++++ 6 files changed, 328 insertions(+) create mode 100644 recipes/fbgemm/LICENSE-asmjit.txt create mode 100644 recipes/fbgemm/LICENSE-cutlass.txt create mode 100644 recipes/fbgemm/LICENSE-fbgemm.txt create mode 100644 recipes/fbgemm/LICENSE-json.txt create mode 100644 recipes/fbgemm/LICENSE-quantize_ops_mx.txt create mode 100644 recipes/fbgemm/meta.yaml diff --git a/recipes/fbgemm/LICENSE-asmjit.txt b/recipes/fbgemm/LICENSE-asmjit.txt new file mode 100644 index 0000000000000..d87dbf9bca28e --- /dev/null +++ b/recipes/fbgemm/LICENSE-asmjit.txt @@ -0,0 +1,17 @@ +Copyright (c) 2008-2024 The AsmJit Authors + +This software is provided 'as-is', without any express or implied +warranty. In no event will the authors be held liable for any damages +arising from the use of this software. + +Permission is granted to anyone to use this software for any purpose, +including commercial applications, and to alter it and redistribute it +freely, subject to the following restrictions: + +1. The origin of this software must not be misrepresented; you must not + claim that you wrote the original software. If you use this software + in a product, an acknowledgment in the product documentation would be + appreciated but is not required. +2. Altered source versions must be plainly marked as such, and must not be + misrepresented as being the original software. +3. This notice may not be removed or altered from any source distribution. diff --git a/recipes/fbgemm/LICENSE-cutlass.txt b/recipes/fbgemm/LICENSE-cutlass.txt new file mode 100644 index 0000000000000..fcce300b79407 --- /dev/null +++ b/recipes/fbgemm/LICENSE-cutlass.txt @@ -0,0 +1,31 @@ +Copyright (c) 2017 - 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +SPDX-License-Identifier: BSD-3-Clause + +``` + Redistribution and use in source and binary forms, with or without + modification, are permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, this + list of conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright notice, + this list of conditions and the following disclaimer in the documentation + and/or other materials provided with the distribution. + + 3. Neither the name of the copyright holder nor the names of its + contributors may be used to endorse or promote products derived from + this software without specific prior written permission. + + THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +``` + +(extracted from cutlass README.md) diff --git a/recipes/fbgemm/LICENSE-fbgemm.txt b/recipes/fbgemm/LICENSE-fbgemm.txt new file mode 100644 index 0000000000000..1c8dd93e668d0 --- /dev/null +++ b/recipes/fbgemm/LICENSE-fbgemm.txt @@ -0,0 +1,30 @@ +BSD License + +For FBGEMM software + +Copyright (c) Meta Platforms, Inc. and affiliates. All rights reserved. + +Redistribution and use in source and binary forms, with or without modification, +are permitted provided that the following conditions are met: + + * Redistributions of source code must retain the above copyright notice, this + list of conditions and the following disclaimer. + + * Redistributions in binary form must reproduce the above copyright notice, + this list of conditions and the following disclaimer in the documentation + and/or other materials provided with the distribution. + + * Neither the name Facebook nor the names of its contributors may be used to + endorse or promote products derived from this software without specific + prior written permission. + +THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED +WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE +DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR +ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES +(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; +LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. diff --git a/recipes/fbgemm/LICENSE-json.txt b/recipes/fbgemm/LICENSE-json.txt new file mode 100644 index 0000000000000..1c1f7a690d815 --- /dev/null +++ b/recipes/fbgemm/LICENSE-json.txt @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2013-2022 Niels Lohmann + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/recipes/fbgemm/LICENSE-quantize_ops_mx.txt b/recipes/fbgemm/LICENSE-quantize_ops_mx.txt new file mode 100644 index 0000000000000..9e841e7a26e4e --- /dev/null +++ b/recipes/fbgemm/LICENSE-quantize_ops_mx.txt @@ -0,0 +1,21 @@ + MIT License + + Copyright (c) Microsoft Corporation. + + Permission is hereby granted, free of charge, to any person obtaining a copy + of this software and associated documentation files (the "Software"), to deal + in the Software without restriction, including without limitation the rights + to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + copies of the Software, and to permit persons to whom the Software is + furnished to do so, subject to the following conditions: + + The above copyright notice and this permission notice shall be included in all + copies or substantial portions of the Software. + + THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + SOFTWARE diff --git a/recipes/fbgemm/meta.yaml b/recipes/fbgemm/meta.yaml new file mode 100644 index 0000000000000..d045428e31686 --- /dev/null +++ b/recipes/fbgemm/meta.yaml @@ -0,0 +1,208 @@ +{% set name = "fbgemm" %} +{% set version = "1.2.0" %} + +package: + name: {{ name|lower }} + version: {{ version }} + +source: + # Need to use git_url instead of tarball due to extensive git submodules + git_url: https://github.com/pytorch/FBGEMM.git + git_rev: v{{ version }} + +build: + number: 0 + skip: true # [py<38] + skip: true # [win] + skip: true # [aarch64] # git_url source requires git on build system, problematic for cross-compilation + +outputs: + - name: fbgemm + build: + script: | + git submodule update --init --recursive + mkdir build && cd build + cmake \ + -DCMAKE_INSTALL_PREFIX=$PREFIX \ + -DCMAKE_PREFIX_PATH=$PREFIX \ + -DCMAKE_BUILD_TYPE=Release \ + -DFBGEMM_LIBRARY_TYPE=shared \ + -DASMJIT_STATIC=OFF \ + -DCPUINFO_LIBRARY_TYPE=shared \ + -DFBGEMM_BUILD_TESTS=OFF \ + -DFBGEMM_BUILD_BENCHMARKS=OFF \ + -DFBGEMM_BUILD_DOCS=OFF \ + -DFBGEMM_BUILD_FBGEMM_GPU=OFF \ + .. + make -j${CPU_COUNT} + make install + string: h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} + run_exports: + - {{ pin_subpackage('fbgemm', max_pin='x.x') }} + requirements: + build: + - {{ compiler('c') }} + - {{ compiler('cxx') }} + - {{ stdlib('c') }} + - cmake + - make + - git + host: + - llvm-openmp # [osx] + - libgomp # [linux] + run: + test: + commands: + - test -f $PREFIX/lib/libfbgemm${SHLIB_EXT} # [unix] + - test -f $PREFIX/include/fbgemm/FbgemmBuild.h # [unix] + + - name: fbgemm-gpu + build: + script: | + cd fbgemm_gpu + {% if cuda_compiler_version == "None" %} + python setup.py --package_variant=cpu --package_channel=release install --prefix=$PREFIX --single-version-externally-managed --record=record.txt + {% else %} + python setup.py --package_variant=cuda --package_channel=release install --prefix=$PREFIX --single-version-externally-managed --record=record.txt + {% endif %} + script_env: + # Set CUDA architectures: starts at 6.0 because code uses __hfma2 (half-precision FMA) + # which requires Compute Capability 6.0+ (Pascal). CC 5.0 (Maxwell) only supports FP16 storage, not arithmetic. + # Compiling with 10.0 failed with error: nvcc fatal: Unsupported gpu architecture 'compute_100' - cuda 12.6 issue ? + - TORCH_CUDA_ARCH_LIST=6.0;7.0;7.5;8.0;8.6;9.0+PTX # [cuda_compiler_version != "None"] + string: cuda{{ cuda_compiler_version | replace('.', '') }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version != "None"] + string: cpu_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version == "None"] + requirements: + build: + - {{ compiler('c') }} + - {{ compiler('cxx') }} + - {{ compiler('cuda') }} # [cuda_compiler_version != "None"] + - {{ stdlib('c') }} + - cmake + - make + - ninja + - git + - python + - pip + - setuptools-git-versioning + - pytorch + - pytorch * *cuda* # [cuda_compiler_version != "None"] + - scikit-build + - tabulate + - jinja2 + - pyyaml + - cuda-cudart-dev # [cuda_compiler_version != "None"] + - cuda-nvrtc-dev # [cuda_compiler_version != "None"] + - cuda-nvtx-dev # [cuda_compiler_version != "None"] + - libcublas-dev # [cuda_compiler_version != "None"] + - libcusolver-dev # [cuda_compiler_version != "None"] + - libcusparse-dev # [cuda_compiler_version != "None"] + - libcurand-dev # [cuda_compiler_version != "None"] + host: + - python + - pip + - setuptools + - setuptools-git-versioning + - wheel + - pytorch + - scikit-build + - numpy + - cuda-version {{ cuda_compiler_version }} # [cuda_compiler_version != "None"] + run: + - python + - pytorch + - numpy + - cuda-version >={{ cuda_compiler_version }} # [cuda_compiler_version != "None"] + - cuda-cudart # [cuda_compiler_version != "None"] + test: + imports: + - fbgemm_gpu + commands: + - python -c "import fbgemm_gpu; print(fbgemm_gpu.__version__)" + + - name: fbgemm-gpu-genai + build: + skip: true # [cuda_compiler_version == "None"] + script: | + cd fbgemm_gpu + python setup.py --package_variant=genai --package_channel=release install --prefix=$PREFIX --single-version-externally-managed --record=record.txt + script_env: + # Set CUDA architectures: starts at 6.0 because genai code uses __hfma2 (half-precision FMA) + # which requires Compute Capability 6.0+ (Pascal). CC 5.0 (Maxwell) only supports FP16 storage, not arithmetic. + # Compiling with 10.0 failed with error: nvcc fatal: Unsupported gpu architecture 'compute_100' - cuda 12.6 issue ? + - TORCH_CUDA_ARCH_LIST=6.0;7.0;7.5;8.0;8.6;9.0+PTX # [cuda_compiler_version != "None"] + string: cuda{{ cuda_compiler_version | replace('.', '') }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version != "None"] + requirements: + build: + - {{ compiler('c') }} + - {{ compiler('cxx') }} + - {{ compiler('cuda') }} # [cuda_compiler_version != "None"] + - {{ stdlib('c') }} + - cmake + - make + - ninja + - git + - python + - pip + - setuptools-git-versioning + - pytorch + - pytorch * *cuda* # [cuda_compiler_version != "None"] + - scikit-build + - tabulate + - jinja2 + - pyyaml + - click + - pyre-extensions + - patchelf # [linux] + - cuda-cccl # [cuda_compiler_version != "None"] + - cuda-cudart-dev # [cuda_compiler_version != "None"] + - cuda-nvrtc-dev # [cuda_compiler_version != "None"] + - cuda-nvtx-dev # [cuda_compiler_version != "None"] + - libcublas-dev # [cuda_compiler_version != "None"] + - libcusolver-dev # [cuda_compiler_version != "None"] + - libcusparse-dev # [cuda_compiler_version != "None"] + - libcurand-dev # [cuda_compiler_version != "None"] + host: + - python + - pip + - setuptools + - setuptools-git-versioning + - wheel + - pytorch + - pytorch * *cuda* # [cuda_compiler_version != "None"] + - scikit-build + - numpy + - cuda-version {{ cuda_compiler_version }} # [cuda_compiler_version != "None"] + run: + - python + - pytorch + - numpy + - cuda-version >={{ cuda_compiler_version }} # [cuda_compiler_version != "None"] + - cuda-cudart # [cuda_compiler_version != "None"] + test: + imports: + - fbgemm_gpu + - fbgemm_gpu.experimental.gen_ai + commands: + - python -c "import fbgemm_gpu; print(fbgemm_gpu.__version__)" + - python -c "import fbgemm_gpu.experimental.gen_ai" + +about: + home: https://github.com/pytorch/FBGEMM + summary: 'FBGEMM GPU kernel libraries for PyTorch' + description: | + FBGEMM (Facebook GEneral Matrix Multiplication) is a low-precision, high-performance matrix-matrix multiplications and convolution library for server-side inference. + The library provides efficient low-precision general matrix multiplication for small batch sizes and support for accuracy-loss minimizing techniques such as row-wise quantization and outlier-aware quantization. FBGEMM also exploits fusion opportunities in order to overcome the unique challenges of matrix multiplication at lower precision with bandwidth-bound operations. + license: BSD-3-Clause AND MIT + license_file: + - LICENSE-fbgemm.txt # LICENSE + - LICENSE-asmjit.txt # external/asmjit/LICENSE.md + - LICENSE-cutlass.txt # external/cutlass/README.md + - LICENSE-json.txt # external/json/LICENSE.MIT + - LICENSE-quantize_ops_mx.txt # fbgemm_gpu/src/quantize_ops/mx/LICENSE + doc_url: https://github.com/pytorch/FBGEMM + dev_url: https://github.com/pytorch/FBGEMM + +extra: + recipe-maintainers: + - das-intensity From c486b95b16056b3367cca0dfb32441a8e3385733 Mon Sep 17 00:00:00 2001 From: Nicholas Dahm Date: Thu, 1 Jan 2026 16:04:36 -0800 Subject: [PATCH 3/6] disabled osx build temporarily --- recipes/fbgemm/meta.yaml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/recipes/fbgemm/meta.yaml b/recipes/fbgemm/meta.yaml index d045428e31686..47f0f82c2e29e 100644 --- a/recipes/fbgemm/meta.yaml +++ b/recipes/fbgemm/meta.yaml @@ -58,6 +58,9 @@ outputs: - name: fbgemm-gpu build: + # TODO: OSX build fails with CMake errors - needs investigation + # OSX should be supported per https://pytorch.org/FBGEMM/fbgemm_gpu/development/BuildInstructions.html + skip: true # [osx] script: | cd fbgemm_gpu {% if cuda_compiler_version == "None" %} From 5b89244aa3b867719402fcfee4ecfe0fc632bd62 Mon Sep 17 00:00:00 2001 From: Nicholas Dahm Date: Sun, 4 Jan 2026 12:39:16 -0800 Subject: [PATCH 4/6] removed redundant skip py<38, submodule update, and command imports --- recipes/fbgemm/meta.yaml | 7 ------- 1 file changed, 7 deletions(-) diff --git a/recipes/fbgemm/meta.yaml b/recipes/fbgemm/meta.yaml index 47f0f82c2e29e..39d65c19d2cd2 100644 --- a/recipes/fbgemm/meta.yaml +++ b/recipes/fbgemm/meta.yaml @@ -12,7 +12,6 @@ source: build: number: 0 - skip: true # [py<38] skip: true # [win] skip: true # [aarch64] # git_url source requires git on build system, problematic for cross-compilation @@ -20,7 +19,6 @@ outputs: - name: fbgemm build: script: | - git submodule update --init --recursive mkdir build && cd build cmake \ -DCMAKE_INSTALL_PREFIX=$PREFIX \ @@ -120,8 +118,6 @@ outputs: test: imports: - fbgemm_gpu - commands: - - python -c "import fbgemm_gpu; print(fbgemm_gpu.__version__)" - name: fbgemm-gpu-genai build: @@ -186,9 +182,6 @@ outputs: imports: - fbgemm_gpu - fbgemm_gpu.experimental.gen_ai - commands: - - python -c "import fbgemm_gpu; print(fbgemm_gpu.__version__)" - - python -c "import fbgemm_gpu.experimental.gen_ai" about: home: https://github.com/pytorch/FBGEMM From deda36f6b0310b0cadf6d8219baabaaa6d1e859e Mon Sep 17 00:00:00 2001 From: Nicholas Dahm Date: Sun, 4 Jan 2026 12:39:40 -0800 Subject: [PATCH 5/6] renamed fbgemm packages to be more clear --- recipes/fbgemm/meta.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/recipes/fbgemm/meta.yaml b/recipes/fbgemm/meta.yaml index 39d65c19d2cd2..d0e2c582afa27 100644 --- a/recipes/fbgemm/meta.yaml +++ b/recipes/fbgemm/meta.yaml @@ -16,7 +16,7 @@ build: skip: true # [aarch64] # git_url source requires git on build system, problematic for cross-compilation outputs: - - name: fbgemm + - name: libfbgemm build: script: | mkdir build && cd build @@ -54,7 +54,7 @@ outputs: - test -f $PREFIX/lib/libfbgemm${SHLIB_EXT} # [unix] - test -f $PREFIX/include/fbgemm/FbgemmBuild.h # [unix] - - name: fbgemm-gpu + - name: fbgemm build: # TODO: OSX build fails with CMake errors - needs investigation # OSX should be supported per https://pytorch.org/FBGEMM/fbgemm_gpu/development/BuildInstructions.html @@ -119,7 +119,7 @@ outputs: imports: - fbgemm_gpu - - name: fbgemm-gpu-genai + - name: fbgemm-genai build: skip: true # [cuda_compiler_version == "None"] script: | From aae5a8fa915586860b54bdd9d7344aeb922be147 Mon Sep 17 00:00:00 2001 From: Nicholas Dahm Date: Sun, 4 Jan 2026 12:41:41 -0800 Subject: [PATCH 6/6] restricted fbgemm arch list to speed up CI while fixing misc PR issues --- recipes/fbgemm/meta.yaml | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/recipes/fbgemm/meta.yaml b/recipes/fbgemm/meta.yaml index d0e2c582afa27..fdc97b7f9f14c 100644 --- a/recipes/fbgemm/meta.yaml +++ b/recipes/fbgemm/meta.yaml @@ -70,7 +70,9 @@ outputs: # Set CUDA architectures: starts at 6.0 because code uses __hfma2 (half-precision FMA) # which requires Compute Capability 6.0+ (Pascal). CC 5.0 (Maxwell) only supports FP16 storage, not arithmetic. # Compiling with 10.0 failed with error: nvcc fatal: Unsupported gpu architecture 'compute_100' - cuda 12.6 issue ? - - TORCH_CUDA_ARCH_LIST=6.0;7.0;7.5;8.0;8.6;9.0+PTX # [cuda_compiler_version != "None"] + # TODO switch back to full arch list before merging + # - TORCH_CUDA_ARCH_LIST=6.0;7.0;7.5;8.0;8.6;9.0+PTX # [cuda_compiler_version != "None"] + - TORCH_CUDA_ARCH_LIST=9.0 # [cuda_compiler_version != "None"] string: cuda{{ cuda_compiler_version | replace('.', '') }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version != "None"] string: cpu_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version == "None"] requirements: @@ -129,7 +131,9 @@ outputs: # Set CUDA architectures: starts at 6.0 because genai code uses __hfma2 (half-precision FMA) # which requires Compute Capability 6.0+ (Pascal). CC 5.0 (Maxwell) only supports FP16 storage, not arithmetic. # Compiling with 10.0 failed with error: nvcc fatal: Unsupported gpu architecture 'compute_100' - cuda 12.6 issue ? - - TORCH_CUDA_ARCH_LIST=6.0;7.0;7.5;8.0;8.6;9.0+PTX # [cuda_compiler_version != "None"] + # TODO switch back to full arch list before merging + # - TORCH_CUDA_ARCH_LIST=6.0;7.0;7.5;8.0;8.6;9.0+PTX # [cuda_compiler_version != "None"] + - TORCH_CUDA_ARCH_LIST=9.0 # [cuda_compiler_version != "None"] string: cuda{{ cuda_compiler_version | replace('.', '') }}_py{{ CONDA_PY }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda_compiler_version != "None"] requirements: build: