Skip to content

Commit f9615de

Browse files
authored
Merge branch 'main' into rapids-26.04
2 parents 96d8831 + 6a9c586 commit f9615de

2 files changed

Lines changed: 8 additions & 50 deletions

File tree

.github/workflows/build-wheel.yml

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,11 @@ jobs:
4646

4747
- name: Install latest rapidsai/sccache
4848
if: ${{ startsWith(inputs.host-platform, 'linux') }}
49+
env:
50+
SCCACHE_TAG: v0.14.0-rapids.3
4951
run: |
50-
curl -fsSL "https://github.com/rapidsai/sccache/releases/latest/download/sccache-$(uname -m)-unknown-linux-musl.tar.gz" \
52+
# curl -fsSL "https://github.com/rapidsai/sccache/releases/latest/download/sccache-$(uname -m)-unknown-linux-musl.tar.gz" \
53+
curl -fsSL "https://github.com/rapidsai/sccache/releases/download/${{ env.SCCACHE_TAG }}/sccache-${{ env.SCCACHE_TAG }}-$(uname -m)-unknown-linux-musl.tar.gz" \
5154
| sudo tar -C /usr/local/bin -xvzf - --wildcards --strip-components=1 -x '*/sccache'
5255
echo "SCCACHE_PATH=/usr/local/bin/sccache" >> "$GITHUB_ENV"
5356
echo "SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE=true" >> "$GITHUB_ENV"
@@ -171,8 +174,11 @@ jobs:
171174

172175
- name: Install latest rapidsai/sccache
173176
if: ${{ startsWith(inputs.host-platform, 'linux') }}
177+
env:
178+
SCCACHE_TAG: v0.14.0-rapids.3
174179
run: |
175-
curl -fsSL "https://github.com/rapidsai/sccache/releases/latest/download/sccache-$(uname -m)-unknown-linux-musl.tar.gz" \
180+
# curl -fsSL "https://github.com/rapidsai/sccache/releases/latest/download/sccache-$(uname -m)-unknown-linux-musl.tar.gz" \
181+
curl -fsSL "https://github.com/rapidsai/sccache/releases/download/${{ env.SCCACHE_TAG }}/sccache-${{ env.SCCACHE_TAG }}-$(uname -m)-unknown-linux-musl.tar.gz" \
176182
| sudo tar -C /usr/local/bin -xvzf - --wildcards --strip-components=1 -x '*/sccache'
177183
echo "SCCACHE_PATH=/usr/local/bin/sccache" >> "$GITHUB_ENV"
178184
echo "SCCACHE_GHA_USE_PREPROCESSOR_CACHE_MODE=true" >> "$GITHUB_ENV"

numba_cuda/numba/cuda/_internal/cuda_fp8.py

Lines changed: 0 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -1823,30 +1823,6 @@ def impl(context, builder, fromty, toty, value):
18231823
_from___nv_fp8_e4m3_to_int8_lower(shim_stream, shim_obj)
18241824

18251825

1826-
def _from___nv_fp8_e4m3_to_int8_lower(shim_stream, shim_obj):
1827-
shim_raw_str = """
1828-
extern "C" __device__ int
1829-
_ZNK13__nv_fp8_e4m3cvcEv_nbst(char &retval, __nv_fp8_e4m3 *self) {
1830-
retval = self->operator char();
1831-
return 0;
1832-
}
1833-
"""
1834-
1835-
@lower_cast(_type_fp8_e4m3, int8)
1836-
def impl(context, builder, fromty, toty, value):
1837-
context.active_code_library.add_linking_file(shim_obj)
1838-
callconv = FunctionCallConv(
1839-
itanium_mangled_name="_ZNK13__nv_fp8_e4m3cvcEv",
1840-
shim_writer=shim_writer,
1841-
shim_code=shim_raw_str,
1842-
)
1843-
sig = signature(toty, fromty)
1844-
return callconv(builder, context, sig, [value])
1845-
1846-
1847-
_from___nv_fp8_e4m3_to_int8_lower(shim_stream, shim_obj)
1848-
1849-
18501826
def _from___nv_fp8_e4m3_to_int16_lower(shim_stream, shim_obj):
18511827
shim_raw_str = """
18521828
extern "C" __device__ int
@@ -2865,30 +2841,6 @@ def impl(context, builder, fromty, toty, value):
28652841
_from___nv_fp8_e8m0_to_int8_lower(shim_stream, shim_obj)
28662842

28672843

2868-
def _from___nv_fp8_e8m0_to_int8_lower(shim_stream, shim_obj):
2869-
shim_raw_str = """
2870-
extern "C" __device__ int
2871-
_ZNK13__nv_fp8_e8m0cvcEv_nbst(char &retval, __nv_fp8_e8m0 *self) {
2872-
retval = self->operator char();
2873-
return 0;
2874-
}
2875-
"""
2876-
2877-
@lower_cast(_type_fp8_e8m0, int8)
2878-
def impl(context, builder, fromty, toty, value):
2879-
context.active_code_library.add_linking_file(shim_obj)
2880-
callconv = FunctionCallConv(
2881-
itanium_mangled_name="_ZNK13__nv_fp8_e8m0cvcEv",
2882-
shim_writer=shim_writer,
2883-
shim_code=shim_raw_str,
2884-
)
2885-
sig = signature(toty, fromty)
2886-
return callconv(builder, context, sig, [value])
2887-
2888-
2889-
_from___nv_fp8_e8m0_to_int8_lower(shim_stream, shim_obj)
2890-
2891-
28922844
def _from___nv_fp8_e8m0_to_int16_lower(shim_stream, shim_obj):
28932845
shim_raw_str = """
28942846
extern "C" __device__ int

0 commit comments

Comments
 (0)